@@ -533,12 +533,12 @@ def _plan(
             ...     batch_source=driver_hourly_stats,
             ... )
             >>> registry_diff, infra_diff, new_infra = fs._plan(RepoContents(
-            ...     data_sources={driver_hourly_stats},
-            ...     feature_views={driver_hourly_stats_view},
-            ...     on_demand_feature_views=set(),
-            ...     request_feature_views=set(),
-            ...     entities={driver},
-            ...     feature_services=set())) # register entity and feature view
+            ...     data_sources=[driver_hourly_stats],
+            ...     feature_views=[driver_hourly_stats_view],
+            ...     on_demand_feature_views=list(),
+            ...     request_feature_views=list(),
+            ...     entities=[driver],
+            ...     feature_services=list())) # register entity and feature view
         """
         # Validate and run inference on all the objects to be registered.
         self._validate_all_feature_views(
 
@@ -9,15 +9,10 @@
 
 def test_push_with_batch():
     push_source = PushSource(
-        name="test",
-        schema=[Field(name="f1", dtype=Float32), Field(name="f2", dtype=Bool)],
-        timestamp_field="event_timestamp",
-        batch_source=BigQuerySource(table="test.test"),
+        name="test", batch_source=BigQuerySource(table="test.test"),
     )
     push_source_proto = push_source.to_proto()
     assert push_source_proto.HasField("batch_source")
-    assert push_source_proto.timestamp_field is not None
-    assert push_source_proto.push_options is not None
 
     push_source_unproto = PushSource.from_proto(push_source_proto)
 
@@ -51,28 +46,17 @@ def test_request_source_primitive_type_to_proto():
 
 def test_hash():
     push_source_1 = PushSource(
-        name="test",
-        schema=[Field(name="f1", dtype=Float32), Field(name="f2", dtype=Bool)],
-        timestamp_field="event_timestamp",
-        batch_source=BigQuerySource(table="test.test"),
+        name="test", batch_source=BigQuerySource(table="test.test"),
     )
     push_source_2 = PushSource(
-        name="test",
-        schema=[Field(name="f1", dtype=Float32), Field(name="f2", dtype=Bool)],
-        timestamp_field="event_timestamp",
-        batch_source=BigQuerySource(table="test.test"),
+        name="test", batch_source=BigQuerySource(table="test.test"),
     )
     push_source_3 = PushSource(
-        name="test",
-        schema=[Field(name="f1", dtype=Float32)],
-        timestamp_field="event_timestamp",
-        batch_source=BigQuerySource(table="test.test"),
+        name="test", batch_source=BigQuerySource(table="test.test2"),
     )
     push_source_4 = PushSource(
         name="test",
-        schema=[Field(name="f1", dtype=Float32)],
-        timestamp_field="event_timestamp",
-        batch_source=BigQuerySource(table="test.test"),
+        batch_source=BigQuerySource(table="test.test2"),
         description="test",
     )