autogluon.timeseries 1.4.1b20250907__py3-none-any.whl → 1.5.1b20260122__py3-none-any.whl

autogluon/timeseries/trainer/model_set_builder.py

@@ -2,7 +2,7 @@ import copy
 import logging
 import re
 from collections import defaultdict
-from typing import Any, Optional, Type, Union
+from typing import Any, Type
 
 from autogluon.common import space
 from autogluon.core import constants
@@ -16,7 +16,7 @@ from autogluon.timeseries.utils.features import CovariateMetadata
 logger = logging.getLogger(__name__)
 
 
-ModelKey = Union[str, Type[AbstractTimeSeriesModel]]
+ModelKey = str | Type[AbstractTimeSeriesModel]
 ModelHyperparameters = dict[str, Any]
 TrainerHyperparameterSpec = dict[ModelKey, list[ModelHyperparameters]]
 
@@ -34,7 +34,7 @@ class TrainableModelSetBuilder:
     def __init__(
         self,
         path: str,
-        freq: Optional[str],
+        freq: str | None,
         prediction_length: int,
         eval_metric: TimeSeriesScorer,
         target: str,
@@ -53,10 +53,10 @@ class TrainableModelSetBuilder:
 
     def get_model_set(
         self,
-        hyperparameters: Union[str, dict, None],
+        hyperparameters: str | dict | None,
         hyperparameter_tune: bool,
-        excluded_model_types: Optional[list[str]],
-        banned_model_names: Optional[list[str]] = None,
+        excluded_model_types: list[str] | None,
+        banned_model_names: list[str] | None = None,
     ) -> list[AbstractTimeSeriesModel]:
         """Resolve hyperparameters and create the requested list of models"""
         models = []
@@ -153,9 +153,9 @@ class HyperparameterBuilder:
 
     def __init__(
         self,
-        hyperparameters: Union[str, dict, None],
+        hyperparameters: str | dict | None,
         hyperparameter_tune: bool,
-        excluded_model_types: Optional[list[str]],
+        excluded_model_types: list[str] | None,
     ):
         self.hyperparameters = hyperparameters
         self.hyperparameter_tune = hyperparameter_tune
@@ -184,7 +184,7 @@ class HyperparameterBuilder:
 
     def _check_and_clean_hyperparameters(
         self,
-        hyperparameters: dict[ModelKey, Union[ModelHyperparameters, list[ModelHyperparameters]]],
+        hyperparameters: dict[ModelKey, ModelHyperparameters | list[ModelHyperparameters]],
     ) -> TrainerHyperparameterSpec:
         """Convert the hyperparameters dictionary to a unified format:
         - Remove 'Model' suffix from model names, if present

autogluon/timeseries/trainer/prediction_cache.py

@@ -1,7 +1,7 @@
 import logging
 from abc import ABC, abstractmethod
 from pathlib import Path
-from typing import Any, Optional
+from typing import Any
 
 from autogluon.common.utils.utils import hash_pandas_df
 from autogluon.core.utils.loaders import load_pkl
@@ -22,16 +22,16 @@ class PredictionCache(ABC):
 
     @abstractmethod
     def get(
-        self, data: TimeSeriesDataFrame, known_covariates: Optional[TimeSeriesDataFrame]
-    ) -> tuple[dict[str, Optional[TimeSeriesDataFrame]], dict[str, float]]:
+        self, data: TimeSeriesDataFrame, known_covariates: TimeSeriesDataFrame | None
+    ) -> tuple[dict[str, TimeSeriesDataFrame | None], dict[str, float]]:
         pass
 
     @abstractmethod
     def put(
         self,
         data: TimeSeriesDataFrame,
-        known_covariates: Optional[TimeSeriesDataFrame],
-        model_pred_dict: dict[str, Optional[TimeSeriesDataFrame]],
+        known_covariates: TimeSeriesDataFrame | None,
+        model_pred_dict: dict[str, TimeSeriesDataFrame | None],
         pred_time_dict: dict[str, float],
     ) -> None:
         pass
@@ -48,7 +48,7 @@ def get_prediction_cache(use_cache: bool, root_path: str) -> PredictionCache:
         return NoOpPredictionCache(root_path=root_path)
 
 
-def compute_dataset_hash(data: TimeSeriesDataFrame, known_covariates: Optional[TimeSeriesDataFrame] = None) -> str:
+def compute_dataset_hash(data: TimeSeriesDataFrame, known_covariates: TimeSeriesDataFrame | None = None) -> str:
     """Compute a unique string that identifies the time series dataset."""
     combined_hash = hash_pandas_df(data) + hash_pandas_df(known_covariates) + hash_pandas_df(data.static_features)
     return combined_hash
@@ -58,15 +58,15 @@ class NoOpPredictionCache(PredictionCache):
     """A dummy (no-op) prediction cache."""
 
     def get(
-        self, data: TimeSeriesDataFrame, known_covariates: Optional[TimeSeriesDataFrame]
-    ) -> tuple[dict[str, Optional[TimeSeriesDataFrame]], dict[str, float]]:
+        self, data: TimeSeriesDataFrame, known_covariates: TimeSeriesDataFrame | None
+    ) -> tuple[dict[str, TimeSeriesDataFrame | None], dict[str, float]]:
         return {}, {}
 
     def put(
         self,
         data: TimeSeriesDataFrame,
-        known_covariates: Optional[TimeSeriesDataFrame],
-        model_pred_dict: dict[str, Optional[TimeSeriesDataFrame]],
+        known_covariates: TimeSeriesDataFrame | None,
+        model_pred_dict: dict[str, TimeSeriesDataFrame | None],
         pred_time_dict: dict[str, float],
     ) -> None:
         pass
@@ -85,16 +85,16 @@ class FileBasedPredictionCache(PredictionCache):
         return Path(self.root_path) / self._cached_predictions_filename
 
     def get(
-        self, data: TimeSeriesDataFrame, known_covariates: Optional[TimeSeriesDataFrame]
-    ) -> tuple[dict[str, Optional[TimeSeriesDataFrame]], dict[str, float]]:
+        self, data: TimeSeriesDataFrame, known_covariates: TimeSeriesDataFrame | None
+    ) -> tuple[dict[str, TimeSeriesDataFrame | None], dict[str, float]]:
         dataset_hash = compute_dataset_hash(data, known_covariates)
         return self._get_cached_pred_dicts(dataset_hash)
 
     def put(
         self,
         data: TimeSeriesDataFrame,
-        known_covariates: Optional[TimeSeriesDataFrame],
-        model_pred_dict: dict[str, Optional[TimeSeriesDataFrame]],
+        known_covariates: TimeSeriesDataFrame | None,
+        model_pred_dict: dict[str, TimeSeriesDataFrame | None],
         pred_time_dict: dict[str, float],
     ) -> None:
         dataset_hash = compute_dataset_hash(data, known_covariates)
@@ -117,7 +117,7 @@ class FileBasedPredictionCache(PredictionCache):
 
     def _get_cached_pred_dicts(
         self, dataset_hash: str
-    ) -> tuple[dict[str, Optional[TimeSeriesDataFrame]], dict[str, float]]:
+    ) -> tuple[dict[str, TimeSeriesDataFrame | None], dict[str, float]]:
         """Load cached predictions for given dataset_hash from disk, if possible.
 
         If loading fails for any reason, empty dicts are returned.
@@ -136,7 +136,7 @@ class FileBasedPredictionCache(PredictionCache):
     def _save_cached_pred_dicts(
         self,
         dataset_hash: str,
-        model_pred_dict: dict[str, Optional[TimeSeriesDataFrame]],
+        model_pred_dict: dict[str, TimeSeriesDataFrame | None],
         pred_time_dict: dict[str, float],
     ) -> None:
         cached_predictions = self._load_cached_predictions()