Skip to content

Commit 17966f5

Browse files
committed
reduce time spent in testing
1 parent 514f46f commit 17966f5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/operators/forecast/test_errors.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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"])
291291
def 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"])
299299
def 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"])
732732
def 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"])
763763
def 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

Comments
 (0)