fix unit tests

logicalclocks · Nov 1, 2023 · 896271a · 896271a
1 parent 6e757a0
commit 896271a
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/python/tests/core/test_feature_view_engine.py b/python/tests/core/test_feature_view_engine.py
@@ -1184,7 +1184,15 @@ def test_read_dir_from_storage_connector(self, mocker):
 
         # Act
         fv_engine._read_dir_from_storage_connector(
-            training_data_obj=td, path="test", read_options=None
+            training_data_obj=td,
+            path="test",
+            read_options=None,
+            with_primary_keys=None,
+            primary_keys=None,
+            with_event_time=None,
+            event_time=None,
+            with_training_helper_columns=None,
+            training_helper_columns=None,
         )
 
         # Assert
@@ -1216,7 +1224,15 @@ def test_read_dir_from_storage_connector_file_not_found(self, mocker):
         # Act
         with pytest.raises(FileNotFoundError) as e_info:
             fv_engine._read_dir_from_storage_connector(
-                training_data_obj=td, path="test", read_options=None
+                training_data_obj=td,
+                path="test",
+                read_options=None,
+                with_primary_keys=None,
+                primary_keys=None,
+                with_event_time=None,
+                event_time=None,
+                with_training_helper_columns=None,
+                training_helper_columns=None,
             )
 
         # Assert