Skip to content

Commit b3ec328

Browse files
committed
adding traceback
1 parent b2d59f6 commit b3ec328

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

ads/opctl/operator/lowcode/forecast/model/ml_forecast.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,8 @@
22

33
# Copyright (c) 2024 Oracle and/or its affiliates.
44
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
5-
import numpy as np
5+
import traceback
6+
67
import pandas as pd
78

89
from ads.common.decorator import runtime_dependency
@@ -164,7 +165,7 @@ def _train_model(self, data_train, data_test, model_kwargs):
164165
self.errors_dict[self.spec.model] = {
165166
"model_name": self.spec.model,
166167
"error": str(e),
167-
"error_trace": traceback.format_exc()
168+
"error_trace": traceback.format_exc(),
168169
}
169170
logger.warn(f"Encountered Error: {e}. Skipping.")
170171
logger.warn(traceback.format_exc())
@@ -173,7 +174,7 @@ def _train_model(self, data_train, data_test, model_kwargs):
173174
def _build_model(self) -> pd.DataFrame:
174175
data_train = self.datasets.get_all_data_long(include_horizon=False)
175176
data_test = self.datasets.get_all_data_long_forecast_horizon()
176-
self.models = dict()
177+
self.models = {}
177178
model_kwargs = self.set_kwargs()
178179
self.forecast_output = ForecastOutput(
179180
confidence_interval_width=self.spec.confidence_interval_width,

0 commit comments

Comments
 (0)