configure tests

codeloop · codeloop · commit 8e23318a8bfa · 2024-08-13T07:20:43.000Z
diff --git a/ads/opctl/operator/lowcode/forecast/model/ml_forecast.py b/ads/opctl/operator/lowcode/forecast/model/ml_forecast.py
@@ -154,7 +154,7 @@ def _train_model(self, data_train, data_test, model_kwargs):
                 )
 
                 self.model_parameters[s_id] = {
-                    "framework": SupportedModels.MLForecast,
+                    "framework": SupportedModels.LGBForecast,
                     **lgb_params,
                 }
 

Original file line number	Diff line number	Diff line change
`@@ -154,7 +154,7 @@ def _train_model(self, data_train, data_test, model_kwargs):`
`154`	`154`	`)`
`155`	`155`
`156`	`156`	`self.model_parameters[s_id] = {`
`157`		`- "framework": SupportedModels.MLForecast,`
	`157`	`+ "framework": SupportedModels.LGBForecast,`
`158`	`158`	`**lgb_params,`
`159`	`159`	`}`
`160`	`160`