Skip to content

Commit e6da4bf

Browse files
authored
Updated forecast operator unit tests (#776)
2 parents a98df6b + 887deea commit e6da4bf

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
@@ -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)