@@ -287,15 +287,15 @@ def setup_artificial_data(tmpdirname, hist_data=None, add_data=None, test_data=N
287
287
return historical_data_path , additional_data_path , test_data_path
288
288
289
289
290
- @pytest .mark .parametrize ("model" , MODELS )
290
+ @pytest .mark .parametrize ("model" , [ "prophet" ] )
291
291
def test_rossman (operator_setup , model ):
292
292
run_operator (
293
293
tmpdirname = operator_setup ,
294
294
model = model ,
295
295
)
296
296
297
297
298
- @pytest .mark .parametrize ("model" , MODELS )
298
+ @pytest .mark .parametrize ("model" , [ "prophet" ] )
299
299
def test_historical_data (operator_setup , model ):
300
300
tmpdirname = operator_setup
301
301
historical_data_path , additional_data_path , _ = setup_artificial_data (tmpdirname )
@@ -728,7 +728,7 @@ def test_smape_error():
728
728
assert result == 0
729
729
730
730
731
- @pytest .mark .parametrize ("model" , MODELS )
731
+ @pytest .mark .parametrize ("model" , [ "prophet" ] )
732
732
def test_pandas_historical_input (operator_setup , model ):
733
733
from ads .opctl .operator .lowcode .forecast .__main__ import operate
734
734
from ads .opctl .operator .lowcode .forecast .model .forecast_datasets import (
@@ -759,7 +759,7 @@ def test_pandas_historical_input(operator_setup, model):
759
759
)
760
760
761
761
762
- @pytest .mark .parametrize ("model" , MODELS )
762
+ @pytest .mark .parametrize ("model" , [ "prophet" ] )
763
763
def test_pandas_additional_input (operator_setup , model ):
764
764
from ads .opctl .operator .lowcode .forecast .__main__ import operate
765
765
from ads .opctl .operator .lowcode .forecast .model .forecast_datasets import (
0 commit comments