From b96a1d14d179cf3814f8fe209f894857b2c412df Mon Sep 17 00:00:00 2001 From: Rob Howley Date: Wed, 30 Oct 2024 15:55:12 -0400 Subject: [PATCH] ignore assignment type checking Signed-off-by: Rob Howley --- sdk/python/feast/feature_server.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/sdk/python/feast/feature_server.py b/sdk/python/feast/feature_server.py index 95e4ce8897..27b7864114 100644 --- 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 )