revert Optional

Author: Lanz-db

runtime/databricks/automl_runtime/forecast/pmdarima/training.py

@@ -15,7 +15,7 @@
 #
 import logging
 import traceback
-from typing import List
+from typing import List, Optional
 
 import pandas as pd
 import pickle
@@ -35,8 +35,8 @@ class ArimaEstimator:
     """
 
     def __init__(self, horizon: int, frequency_unit: str, metric: str, seasonal_periods: List[int],
-                 num_folds: int = 20, max_steps: int = 150, exogenous_cols: List[str] | None = None,
-                 split_cutoff: pd.Timestamp | None = None) -> None:
+                 num_folds: int = 20, max_steps: int = 150, exogenous_cols: Optional[List[str]] = None,
+                 split_cutoff: Optional[pd.Timestamp] = None) -> None:
         """
         :param horizon: Number of periods to forecast forward
         :param frequency_unit: Frequency of the time series

runtime/databricks/automl_runtime/forecast/prophet/forecast.py

@@ -92,7 +92,7 @@ def __init__(self, horizon: int, frequency_unit: str, metric: str, interval_widt
                  max_eval: int = 10, trial_timeout: int = None,
                  random_state: int = 0, is_parallel: bool = True,
                  regressors = None, 
-                 split_cutoff: pd.Timestamp | None = None, **prophet_kwargs) -> None:
+                 split_cutoff: Optional[pd.Timestamp] = None, **prophet_kwargs) -> None:
         """
         Initialization