From f6d7726e8abf0436601510e3969eba1c356b50f8 Mon Sep 17 00:00:00 2001 From: Prashant Sankhla Date: Fri, 26 Jul 2024 11:16:50 +0530 Subject: [PATCH] fixed target removal if all unique --- ads/opctl/operator/lowcode/forecast/model/arima.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ads/opctl/operator/lowcode/forecast/model/arima.py b/ads/opctl/operator/lowcode/forecast/model/arima.py index 0b3c02768..549ffff51 100644 --- a/ads/opctl/operator/lowcode/forecast/model/arima.py +++ b/ads/opctl/operator/lowcode/forecast/model/arima.py @@ -67,7 +67,9 @@ def _train_model(self, i, s_id, df, model_kwargs): self.forecast_output.init_series_output(series_id=s_id, data_at_series=df) # If trend is constant, remove constant columns if "trend" not in model_kwargs or model_kwargs["trend"] == "c": - self.constant_cols[s_id] = df.columns[df.nunique() == 1] + self.constant_cols[s_id] = list(df.columns[df.nunique() == 1]) + if target in self.constant_cols[s_id]: + self.constant_cols[s_id].remove(target) df = df.drop(columns=self.constant_cols[s_id]) # format the dataframe for this target. Dropping NA on target[df] will remove all future data