fix test

achals · achals · commit 6d4006341d37 · 2021-06-28T11:38:09.000-07:00
Signed-off-by: Achal Shah &lt;achals@gmail.com&gt;
diff --git a/sdk/python/feast/feature_view.py b/sdk/python/feast/feature_view.py
@@ -175,6 +175,8 @@ def to_proto(self) -> FeatureViewProto:
             ttl_duration = Duration()
             ttl_duration.FromTimedelta(self.ttl)
 
+        print(f"Stream soruce: {self.stream_source}, {type(self.stream_source)}")
+
         spec = FeatureViewSpecProto(
             name=self.name,
             entities=self.entities,
@@ -205,6 +207,9 @@ def from_proto(cls, feature_view_proto: FeatureViewProto):
         """
 
         _input = DataSource.from_proto(feature_view_proto.spec.batch_source)
+        stream_source = DataSource.from_proto(feature_view_proto.spec.stream_source)\
+            if feature_view_proto.spec.HasField("stream_source") \
+            else None
         feature_view = cls(
             name=feature_view_proto.spec.name,
             entities=[entity for entity in feature_view_proto.spec.entities],
@@ -226,11 +231,7 @@ def from_proto(cls, feature_view_proto: FeatureViewProto):
             ),
             input=_input,
             batch_source=_input,
-            stream_source=(
-                feature_view_proto.spec.stream_source
-                if feature_view_proto.spec.stream_source is not None
-                else None
-            ),
+            stream_source=stream_source,
         )
 
         feature_view.created_timestamp = feature_view_proto.meta.created_timestamp
