fix automlx and theta forecaster bug (#1344)

Author: skjoracle

ads/opctl/operator/lowcode/common/utils.py

@@ -419,6 +419,16 @@ def normalize_frequency(freq: str) -> str:
     if freq.startswith("W-"):
         return "W"
 
+    if freq.startswith("YS-"):
+        # Extract the month part after "YS-" (e.g., "JAN", "FEB", "MAR")
+        month_anchor = freq[3:]
+        valid_months = {"JAN", "FEB", "MAR", "APR", "MAY", "JUN", "JUL", "AUG", "SEP", "OCT", "NOV", "DEC"}
+
+        if month_anchor in valid_months:
+            return "Y"  # Treat as standard year frequency
+        else:
+            return freq
+
     # Handle month start/end frequencies
     freq_mapping = {
         "MS": "M",  # Month Start -> Month End

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

@@ -185,9 +185,9 @@ def _build_model(self) -> pd.DataFrame:
                 self.forecast_output.populate_series_output(
                     series_id=s_id,
                     fit_val=fitted_values,
-                    forecast_val=summary_frame[target],
-                    upper_bound=summary_frame[f"{target}_ci_upper"],
-                    lower_bound=summary_frame[f"{target}_ci_lower"],
+                    forecast_val=summary_frame[target].values,
+                    upper_bound=summary_frame[f"{target}_ci_upper"].values,
+                    lower_bound=summary_frame[f"{target}_ci_lower"].values,
                 )
 
                 self.model_parameters[s_id] = {