historical_field_mappings making pyling happy

michelle-rascati-sp · michelle-rascati-sp · commit af8fc8530b26 · 2022-01-27T14:21:32.000-06:00
diff --git a/sdk/python/feast/feature_store.py b/sdk/python/feast/feature_store.py
@@ -70,7 +70,7 @@
 from feast.protos.feast.serving.ServingService_pb2 import (
     FieldStatus,
     GetOnlineFeaturesResponse,
-)
+    FieldStatusValue)
 from feast.protos.feast.types.EntityKey_pb2 import EntityKey as EntityKeyProto
 from feast.protos.feast.types.Value_pb2 import RepeatedValue, Value
 from feast.registry import Registry
@@ -1349,7 +1349,7 @@ def _read_from_online_store(
         provider: Provider,
         requested_features: List[str],
         table: FeatureView,
-    ) -> List[Tuple[List[Timestamp], List["FieldStatus.ValueType"], List[Value]]]:
+    ) -> List[Tuple[List[Timestamp], List[FieldStatusValue], List[Value]]]:
         """ Read and process data from the OnlineStore for a given FeatureView.
 
             This method guarentees that the order of the data in each element of the
@@ -1404,7 +1404,7 @@ def _read_from_online_store(
     def _populate_response_from_feature_data(
         feature_data: Iterable[
             Tuple[
-                Iterable[Timestamp], Iterable["FieldStatus.ValueType"], Iterable[Value]
+                Iterable[Timestamp], Iterable[FieldStatusValue], Iterable[Value]
             ]
         ],
         indexes: Iterable[Iterable[int]],
