added trace in error.json

prasankh · prasankh · commit 3d7f8de816c0 · 2024-08-21T14:33:10.000+05:30
diff --git a/ads/opctl/operator/lowcode/forecast/model/arima.py b/ads/opctl/operator/lowcode/forecast/model/arima.py
@@ -126,7 +126,10 @@ def _train_model(self, i, s_id, df, model_kwargs):
 
             logger.debug("===========Done===========")
         except Exception as e:
-            self.errors_dict[s_id] = {"model_name": self.spec.model, "error": str(e)}
+            self.errors_dict[s_id] = {
+                "model_name": self.spec.model,
+                "error": str(e),
+                "error_trace": traceback.format_exc()}
             logger.warn(f"Encountered Error: {e}. Skipping.")
             logger.warn(traceback.format_exc())
 
diff --git a/ads/opctl/operator/lowcode/forecast/model/automlx.py b/ads/opctl/operator/lowcode/forecast/model/automlx.py
@@ -170,6 +170,7 @@ def _build_model(self) -> pd.DataFrame:
                 self.errors_dict[s_id] = {
                     "model_name": self.spec.model,
                     "error": str(e),
+                    "error_trace": traceback.format_exc()
                 }
                 logger.warn(f"Encountered Error: {e}. Skipping.")
                 logger.warn(traceback.format_exc())
diff --git a/ads/opctl/operator/lowcode/forecast/model/autots.py b/ads/opctl/operator/lowcode/forecast/model/autots.py
@@ -208,6 +208,7 @@ def _build_model(self) -> pd.DataFrame:
                 self.errors_dict[s_id] = {
                     "model_name": self.spec.model,
                     "error": str(e),
+                    "error_trace": traceback.format_exc()
                 }
             logger.warn(f"Encountered Error: {e}. Skipping.")
             logger.warn(traceback.format_exc())
diff --git a/ads/opctl/operator/lowcode/forecast/model/ml_forecast.py b/ads/opctl/operator/lowcode/forecast/model/ml_forecast.py
@@ -164,6 +164,7 @@ def _train_model(self, data_train, data_test, model_kwargs):
             self.errors_dict[self.spec.model] = {
                 "model_name": self.spec.model,
                 "error": str(e),
+                "error_trace": traceback.format_exc()
             }
             logger.warn(f"Encountered Error: {e}. Skipping.")
             logger.warn(traceback.format_exc())
diff --git a/ads/opctl/operator/lowcode/forecast/model/neuralprophet.py b/ads/opctl/operator/lowcode/forecast/model/neuralprophet.py
@@ -217,7 +217,11 @@ def _train_model(self, i, s_id, df, model_kwargs):
 
             logger.debug("===========Done===========")
         except Exception as e:
-            self.errors_dict[s_id] = {"model_name": self.spec.model, "error": str(e)}
+            self.errors_dict[s_id] = {
+                "model_name": self.spec.model,
+                "error": str(e),
+                "error_trace": traceback.format_exc()
+            }
             logger.warn(traceback.format_exc())
             raise e
 
diff --git a/ads/opctl/operator/lowcode/forecast/model/prophet.py b/ads/opctl/operator/lowcode/forecast/model/prophet.py
@@ -130,6 +130,7 @@ def _train_model(self, i, series_id, df, model_kwargs):
             self.errors_dict[series_id] = {
                 "model_name": self.spec.model,
                 "error": str(e),
+                "error_trace": traceback.format_exc()
             }
             logger.warn(f"Encountered Error: {e}. Skipping.")
             logger.warn(traceback.format_exc())

Original file line number	Diff line number	Diff line change
`@@ -170,6 +170,7 @@ def _build_model(self) -> pd.DataFrame:`
`170`	`170`	`self.errors_dict[s_id] = {`
`171`	`171`	`"model_name": self.spec.model,`
`172`	`172`	`"error": str(e),`
	`173`	`+ "error_trace": traceback.format_exc()`
`173`	`174`	`}`
`174`	`175`	`logger.warn(f"Encountered Error: {e}. Skipping.")`
`175`	`176`	`logger.warn(traceback.format_exc())`
Original file line number	Diff line number	Diff line change
`@@ -208,6 +208,7 @@ def _build_model(self) -> pd.DataFrame:`
`208`	`208`	`self.errors_dict[s_id] = {`
`209`	`209`	`"model_name": self.spec.model,`
`210`	`210`	`"error": str(e),`
	`211`	`+ "error_trace": traceback.format_exc()`
`211`	`212`	`}`
`212`	`213`	`logger.warn(f"Encountered Error: {e}. Skipping.")`
`213`	`214`	`logger.warn(traceback.format_exc())`
Original file line number	Diff line number	Diff line change
`@@ -164,6 +164,7 @@ def _train_model(self, data_train, data_test, model_kwargs):`
`164`	`164`	`self.errors_dict[self.spec.model] = {`
`165`	`165`	`"model_name": self.spec.model,`
`166`	`166`	`"error": str(e),`
	`167`	`+ "error_trace": traceback.format_exc()`
`167`	`168`	`}`
`168`	`169`	`logger.warn(f"Encountered Error: {e}. Skipping.")`
`169`	`170`	`logger.warn(traceback.format_exc())`
Original file line number	Diff line number	Diff line change
`@@ -130,6 +130,7 @@ def _train_model(self, i, series_id, df, model_kwargs):`
`130`	`130`	`self.errors_dict[series_id] = {`
`131`	`131`	`"model_name": self.spec.model,`
`132`	`132`	`"error": str(e),`
	`133`	`+ "error_trace": traceback.format_exc()`
`133`	`134`	`}`
`134`	`135`	`logger.warn(f"Encountered Error: {e}. Skipping.")`
`135`	`136`	`logger.warn(traceback.format_exc())`