Skip to content

Commit 4c291fa

Browse files
committed
added model_list to speed up tests
1 parent 43974af commit 4c291fa

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/operators/forecast/test_datasets.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,8 @@ def test_load_datasets(model, data_details):
134134
yaml_i["spec"]["model_kwargs"] = {"model_list": "superfast"}
135135
if model == "automlx":
136136
yaml_i["spec"]["model_kwargs"] = {"time_budget": 2}
137+
if model == "auto-select":
138+
yaml_i["spec"]["model_kwargs"] = {"model_list": ['arima', 'mlforecast']}
137139

138140
run(yaml_i, backend="operator.local", debug=False)
139141
subprocess.run(f"ls -a {output_data_path}", shell=True)

0 commit comments

Comments
 (0)