Skip to content

Commit 5de5f6b

Browse files
authored
Update automlx.py
1 parent eee5a2b commit 5de5f6b

File tree

1 file changed

+2
-1
lines changed
  • ads/opctl/operator/lowcode/forecast/model

1 file changed

+2
-1
lines changed

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def preprocess(self, data): # TODO: re-use self.le for explanations
8080
def _build_model(self) -> pd.DataFrame:
8181
import logging
8282

83-
from automlx import Pipeline, init
83+
from automlx import Pipeline, init, shutdown
8484

8585
full_data_dict = self.datasets.get_data_by_series()
8686

@@ -193,6 +193,7 @@ def _build_model(self) -> pd.DataFrame:
193193

194194
logger.debug("===========Forecast Generated===========")
195195

196+
shutdown()
196197
return self.forecast_output.get_forecast_long()
197198

198199
def _generate_report(self):

0 commit comments

Comments
 (0)