From 896271a7e5d3519200adf6b27f47c2b77b785e40 Mon Sep 17 00:00:00 2001
From: davitbzh <davit.bzhalava@gmail.com>
Date: Wed, 1 Nov 2023 18:29:53 +0100
Subject: [PATCH] fix unit tests

---
 python/tests/core/test_feature_view_engine.py | 20 +++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/python/tests/core/test_feature_view_engine.py b/python/tests/core/test_feature_view_engine.py
index cad184c314..6d8a70d1e4 100644
--- 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