Skip to content

Commit 00f9be6

Browse files
authored
Merge branch 'main' into spavlusieva-patch-1
2 parents d38669f + e6da4bf commit 00f9be6

File tree

6 files changed

+1212
-33
lines changed

6 files changed

+1212
-33
lines changed

tests/operators/forecast/test_errors.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -559,10 +559,10 @@ def test_all_series_failure(model):
559559
historical_data_path=historical_data_path,
560560
additional_data_path=additional_data_path,
561561
)
562-
562+
preprocessing_steps = {"missing_value_imputation": True, "outlier_treatment": False}
563563
yaml_i["spec"]["model"] = model
564564
yaml_i['spec']['horizon'] = 10
565-
yaml_i['spec']['preprocessing'] = True
565+
yaml_i['spec']['preprocessing'] = preprocessing_steps
566566
if yaml_i["spec"].get("additional_data") is not None and model != "autots":
567567
yaml_i["spec"]["generate_explanations"] = True
568568
if model == "autots":
@@ -627,9 +627,9 @@ def test_arima_automlx_errors(operator_setup, model):
627627
any supported types according to the casting rule ''safe''
628628
Added label encoding before passing data to explainer
629629
"""
630-
630+
preprocessing_steps = {"missing_value_imputation": True, "outlier_treatment": False}
631631
yaml_i['spec']['horizon'] = 10
632-
yaml_i['spec']['preprocessing'] = True
632+
yaml_i['spec']['preprocessing'] = preprocessing_steps
633633
yaml_i['spec']['generate_explanations'] = True
634634
yaml_i['spec']['model'] = model
635635

0 commit comments

Comments
 (0)