@@ -287,15 +287,15 @@ def setup_artificial_data(tmpdirname, hist_data=None, add_data=None, test_data=N
287287 return historical_data_path , additional_data_path , test_data_path
288288
289289
290- @pytest .mark .parametrize ("model" , MODELS )
290+ @pytest .mark .parametrize ("model" , [ "prophet" ] )
291291def test_rossman (operator_setup , model ):
292292 run_operator (
293293 tmpdirname = operator_setup ,
294294 model = model ,
295295 )
296296
297297
298- @pytest .mark .parametrize ("model" , MODELS )
298+ @pytest .mark .parametrize ("model" , [ "prophet" ] )
299299def test_historical_data (operator_setup , model ):
300300 tmpdirname = operator_setup
301301 historical_data_path , additional_data_path , _ = setup_artificial_data (tmpdirname )
@@ -728,7 +728,7 @@ def test_smape_error():
728728 assert result == 0
729729
730730
731- @pytest .mark .parametrize ("model" , MODELS )
731+ @pytest .mark .parametrize ("model" , [ "prophet" ] )
732732def test_pandas_historical_input (operator_setup , model ):
733733 from ads .opctl .operator .lowcode .forecast .__main__ import operate
734734 from ads .opctl .operator .lowcode .forecast .model .forecast_datasets import (
@@ -759,7 +759,7 @@ def test_pandas_historical_input(operator_setup, model):
759759 )
760760
761761
762- @pytest .mark .parametrize ("model" , MODELS )
762+ @pytest .mark .parametrize ("model" , [ "prophet" ] )
763763def test_pandas_additional_input (operator_setup , model ):
764764 from ads .opctl .operator .lowcode .forecast .__main__ import operate
765765 from ads .opctl .operator .lowcode .forecast .model .forecast_datasets import (
0 commit comments