mark spark run

Author: JQGoh

nixtla_tests/nixtla_client/test_finetune_and_forecast.py

@@ -118,6 +118,7 @@ def test_ray_finetune_model(self, custom_client, ray_df):
         check_finetuned_model(custom_client, ray_df, model_ids_object.model_id2)
 
     @pytest.mark.distributed_run
+    @pytest.mark.spark_run
     def test_spark_finetune_model(self, custom_client, spark_df):
         check_finetuned_model(custom_client, spark_df, model_ids_object.model_id2)
 

nixtla_tests/nixtla_client/test_spark.py

@@ -9,7 +9,10 @@
 from nixtla_tests.helpers.checks import check_forecast_x_dataframe_diff_cols
 from nixtla_tests.helpers.checks import check_quantiles
 
-pytestmark = pytest.mark.distributed_run
+pytestmark = [
+    pytest.mark.distributed_run,
+    pytest.mark.spark_run
+]
 
 def test_quantiles(nixtla_test_client, spark_df):
     check_quantiles(nixtla_test_client, spark_df, id_col="unique_id", time_col="ds")