autots future regressors

Author: ahosler

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

@@ -130,14 +130,17 @@ def _build_model(self) -> pd.DataFrame:
         additional_regressors = list(
             set(full_data_indexed.columns) - {self.original_target_column}
         )
-        future_regressor = full_data_indexed.reset_index().pivot(
-            index=self.spec.datetime_column.name,
-            columns=ForecastOutputColumns.SERIES,
-            values=additional_regressors,
-        )
-
-        future_reg = future_regressor[: -self.spec.horizon]
-        regr_fcst = future_regressor[-self.spec.horizon :]
+        if len(additional_regressors) > 1:
+            future_regressor = full_data_indexed.reset_index().pivot(
+                index=self.spec.datetime_column.name,
+                columns=ForecastOutputColumns.SERIES,
+                values=additional_regressors,
+            )
+            future_reg = future_regressor[: -self.spec.horizon]
+            regr_fcst = future_regressor[-self.spec.horizon :]
+        else:
+            future_reg = None
+            regr_fcst = None
 
         if self.loaded_models is None:
             model = model.fit(

tests/operators/forecast/test_datasets.py

@@ -43,14 +43,14 @@
     # "UberTLCDataset",
     #     "WeatherDataset",
     #     "WineDataset",
-    #     "WoolyDataset",
+    "WoolyDataset",
 ]
 
 MODELS = [
-    # "arima",
-    # "automlx",
-    # "prophet",
-    # "neuralprophet",
+    "arima",
+    "automlx",
+    "prophet",
+    "neuralprophet",
     "autots",
     # "auto",
 ]