fix: Support push sources in stream feature views

achals · achals · commit 1b8b42dcfda0 · 2022-05-16T11:13:05.000-07:00
Signed-off-by: Achal Shah &lt;achals@gmail.com&gt;
diff --git a/sdk/python/feast/stream_feature_view.py b/sdk/python/feast/stream_feature_view.py
@@ -7,10 +7,7 @@
 from feast.field import Field
 from feast.protos.feast.core.DataSource_pb2 import DataSource as DataSourceProto
 
-SUPPORTED_STREAM_SOURCES = {
-    "KafkaSource",
-    "KinesisSource",
-}
+SUPPORTED_STREAM_SOURCES = {"KafkaSource", "KinesisSource", "PushSource"}
 
 
 class StreamFeatureView(FeatureView):
diff --git a/sdk/python/tests/unit/test_feature_views.py b/sdk/python/tests/unit/test_feature_views.py
@@ -2,6 +2,7 @@
 
 import pytest
 
+from feast import PushSource
 from feast.batch_feature_view import BatchFeatureView
 from feast.data_format import AvroFormat
 from feast.data_source import KafkaSource
@@ -50,12 +51,22 @@ def test_create_stream_feature_view():
         batch_source=FileSource(path="some path"),
     )
     StreamFeatureView(
-        name="test batch feature view",
+        name="test kafka stream feature view",
         entities=[],
         ttl=timedelta(days=30),
         source=stream_source,
     )
 
+    push_source = PushSource(
+        name="push source", batch_source=FileSource(path="some path")
+    )
+    StreamFeatureView(
+        name="test push source feature view",
+        entities=[],
+        ttl=timedelta(days=30),
+        source=push_source,
+    )
+
     with pytest.raises(ValueError):
         StreamFeatureView(
             name="test batch feature view", entities=[], ttl=timedelta(days=30)
