[Tests] Fix spark merger tests (#3646)

mlrun · May 28, 2023 · bfd8b97 · bfd8b97
1 parent 50d2733
commit bfd8b97
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/tests/system/feature_store/test_spark_engine.py b/tests/system/feature_store/test_spark_engine.py
@@ -1362,7 +1362,7 @@ def test_get_offline_features_with_filter_and_indexes(self, timestamp_key):
             target=target,
             query="bad>6 and bad<8",
             engine="spark",
-            run_config=fstore.RunConfig(local=self.run_local),
+            run_config=fstore.RunConfig(local=self.run_local, kind="remote-spark"),
             spark_service=self.spark_service,
         )
         resp_df = resp.to_dataframe()
@@ -1432,7 +1432,7 @@ def test_get_offline_features_with_spark_engine(self, passthrough, target_type):
             fv_name,
             target=target,
             query="bad>6 and bad<8",
-            run_config=fstore.RunConfig(local=self.run_local),
+            run_config=fstore.RunConfig(local=self.run_local, kind="remote-spark"),
             engine="spark",
             spark_service=self.spark_service,
         )
@@ -1929,7 +1929,7 @@ def test_relation_join(self, join_type, with_indexes):
             vector,
             target=target,
             with_indexes=with_indexes,
-            run_config=fstore.RunConfig(local=self.run_local),
+            run_config=fstore.RunConfig(local=self.run_local, kind="remote-spark"),
             engine="spark",
             spark_service=self.spark_service,
             join_type=join_type,
@@ -1959,7 +1959,7 @@ def test_relation_join(self, join_type, with_indexes):
             vector,
             target=target,
             with_indexes=with_indexes,
-            run_config=fstore.RunConfig(local=self.run_local),
+            run_config=fstore.RunConfig(local=self.run_local, kind="remote-spark"),
             engine="spark",
             spark_service=self.spark_service,
             join_type=join_type,
@@ -1985,7 +1985,7 @@ def test_relation_join(self, join_type, with_indexes):
             vector,
             target=target,
             with_indexes=with_indexes,
-            run_config=fstore.RunConfig(local=self.run_local),
+            run_config=fstore.RunConfig(local=self.run_local, kind="remote-spark"),
             engine="spark",
             spark_service=self.spark_service,
             join_type=join_type,
@@ -2007,7 +2007,7 @@ def test_relation_join(self, join_type, with_indexes):
             vector,
             target=target,
             with_indexes=with_indexes,
-            run_config=fstore.RunConfig(local=self.run_local),
+            run_config=fstore.RunConfig(local=self.run_local, kind="remote-spark"),
             engine="spark",
             spark_service=self.spark_service,
             join_type=join_type,
@@ -2034,7 +2034,7 @@ def test_relation_join(self, join_type, with_indexes):
             vector,
             target=target,
             with_indexes=with_indexes,
-            run_config=fstore.RunConfig(local=self.run_local),
+            run_config=fstore.RunConfig(local=self.run_local, kind="remote-spark"),
             engine="spark",
             spark_service=self.spark_service,
             join_type=join_type,
@@ -2117,7 +2117,7 @@ def test_relation_asof_join(self, with_indexes):
             vector,
             target=target,
             with_indexes=with_indexes,
-            run_config=fstore.RunConfig(local=self.run_local),
+            run_config=fstore.RunConfig(local=self.run_local, kind="remote-spark"),
             engine="spark",
             spark_service=self.spark_service,
             order_by=["n"],
@@ -2202,7 +2202,7 @@ def test_as_of_join_result(self):
         resp = fstore.get_offline_features(
             vec_for_spark,
             engine="spark",
-            run_config=fstore.RunConfig(local=self.run_local),
+            run_config=fstore.RunConfig(local=self.run_local, kind="remote-spark"),
             spark_service=self.spark_service,
             target=target,
         )