ignore assignment type checking

Signed-off-by: Rob Howley <howley.robert@gmail.com>
robhowley · Oct 30, 2024 · b96a1d1 · b96a1d1
1 parent e26a9de
commit b96a1d1
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/sdk/python/feast/feature_server.py b/sdk/python/feast/feature_server.py
@@ -3,7 +3,7 @@
 import time
 import traceback
 from contextlib import asynccontextmanager
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional
 
 import pandas as pd
 import psutil
@@ -17,7 +17,7 @@
 from pydantic import BaseModel, Field
 
 import feast
-from feast import FeatureView, StreamFeatureView, proto_json, utils
+from feast import proto_json, utils
 from feast.constants import DEFAULT_FEATURE_SERVER_REGISTRY_TTL
 from feast.data_source import PushMode
 from feast.errors import (
@@ -224,13 +224,11 @@ def write_to_online_store(request: WriteToFeatureStoreRequest):
         feature_view_name = request.feature_view_name
         allow_registry_cache = request.allow_registry_cache
         try:
-            feature_view: Union[StreamFeatureView, FeatureView] = (
-                store.get_stream_feature_view(
-                    feature_view_name, allow_registry_cache=allow_registry_cache
-                )
+            feature_view = store.get_stream_feature_view(  # type: ignore
+                feature_view_name, allow_registry_cache=allow_registry_cache
             )
         except FeatureViewNotFoundException:
-            feature_view = store.get_feature_view(
+            feature_view = store.get_feature_view(  # type: ignore
                 feature_view_name, allow_registry_cache=allow_registry_cache
             )