Fix issues

kevjumba · kevjumba · commit 710169d4d097 · 2022-06-02T15:09:05.000-07:00
Signed-off-by: Kevin Zhang &lt;kzhang@tecton.ai&gt;
diff --git a/sdk/python/feast/repo_operations.py b/sdk/python/feast/repo_operations.py
@@ -147,7 +147,7 @@ def parse_repo(repo_root: Path) -> RepoContents:
                 ):
                     batch_source_dep = obj.stream_source.batch_source
                     # Don't add if the push source's batch source is a duplicate of an existing batch source
-                    if batch_source_dep not in data_sources_set:
+                    if batch_source_dep and batch_source_dep not in data_sources_set:
                         res.data_sources.append(batch_source_dep)
             elif isinstance(obj, Entity) and not any(
                 (obj is entity) for entity in res.entities
diff --git a/sdk/python/tests/integration/registration/test_stream_feature_view_apply.py b/sdk/python/tests/integration/registration/test_stream_feature_view_apply.py
@@ -62,7 +62,7 @@ def simple_sfv(df):
     assert entities[0] == entity
 
     features = fs.get_online_features(
-        features=["simple_sfv:dummy_field",], entity_rows=[{"test_key": 1001}],
+        features=["simple_sfv:dummy_field"], entity_rows=[{"test_key": 1001}],
     ).to_dict(include_event_timestamps=True)
 
     assert "test_key" in features
