fix tests

Signed-off-by: Achal Shah <achals@gmail.com>
feast-dev · Apr 11, 2022 · dc06cc2 · dc06cc2
1 parent 65e808b
commit dc06cc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/tests/unit/test_data_sources.py b/sdk/python/tests/unit/test_data_sources.py
@@ -12,8 +12,8 @@ def test_push_with_batch():
         batch_source=BigQuerySource(table="test.test"),
     )
     push_source_proto = push_source.to_proto()
+    assert push_source_proto.HasField("batch_source")
     assert push_source_proto.push_options is not None
-    assert push_source_proto.push_options.HasField("batch_source")
 
     push_source_unproto = PushSource.from_proto(push_source_proto)