fix tests

Signed-off-by: Achal Shah <[email protected]>
feast-dev · Apr 11, 2022 · 67c41b4 · 67c41b4
1 parent dc06cc2
commit 67c41b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/feast/data_source.py b/sdk/python/feast/data_source.py
@@ -704,7 +704,7 @@ def from_proto(data_source: DataSourceProto):
         for key, val in schema_pb.items():
             schema[key] = ValueType(val)
 
-        assert data_source.push_options.HasField("batch_source")
+        assert data_source.HasField("batch_source")
         batch_source = DataSource.from_proto(data_source.batch_source)
 
         return PushSource(