configure tests

Author: codeloop

tests/operators/forecast/test_datasets.py

@@ -32,7 +32,7 @@
     "prophet",
     "neuralprophet",
     "autots",
-    "mlforecast",
+    "lgbforecast",
     "auto-select",
 ]
 
@@ -135,15 +135,15 @@ def test_load_datasets(model, data_details):
         if model == "automlx":
             yaml_i["spec"]["model_kwargs"] = {"time_budget": 2}
         if model == "auto-select":
-            yaml_i["spec"]["model_kwargs"] = {"model_list": ['prophet', 'arima', 'mlforecast']}
+            yaml_i["spec"]["model_kwargs"] = {"model_list": ['prophet', 'arima', 'lgbforecast']}
             if dataset_name == f'{DATASET_PREFIX}dataset4.csv':
                 pytest.skip("Skipping dataset4 with auto-select")  # todo:// ODSC-58584
 
         run(yaml_i, backend="operator.local", debug=False)
         subprocess.run(f"ls -a {output_data_path}", shell=True)
         if yaml_i["spec"]["generate_explanations"] and model not in [
             "automlx",
-            "mlforecast",
+            "lgbforecast",
             "auto-select"
         ]:
             verify_explanations(

tests/operators/forecast/test_errors.py

@@ -141,7 +141,7 @@
     "prophet",
     "neuralprophet",
     "autots",
-    "mlforecast",
+    "lgbforecast",
     # "auto",
 ]
 
@@ -687,7 +687,7 @@ def test_arima_automlx_errors(operator_setup, model):
                 in error_content["13"]["error"]
             ), "Error message mismatch"
 
-    if model not in ["autots", "automlx", "mlforecast"]:
+    if model not in ["autots", "automlx", "lgbforecast"]:
         global_fn = f"{tmpdirname}/results/global_explanation.csv"
         assert os.path.exists(
             global_fn