autogluon.timeseries 1.4.1b20250820__py3-none-any.whl → 1.4.1b20250902__py3-none-any.whl

autogluon/timeseries/transforms/target_scaler.py

@@ -1,4 +1,4 @@
-from typing import Literal, Optional, Protocol, Tuple, Union, overload
+from typing import Literal, Optional, Protocol, Union, overload
 
 import numpy as np
 import pandas as pd
@@ -30,7 +30,7 @@ class LocalTargetScaler(TargetScaler):
         self.loc: Optional[pd.Series] = None
         self.scale: Optional[pd.Series] = None
 
-    def _compute_loc_scale(self, target_series: pd.Series) -> Tuple[Optional[pd.Series], Optional[pd.Series]]:
+    def _compute_loc_scale(self, target_series: pd.Series) -> tuple[Optional[pd.Series], Optional[pd.Series]]:
         raise NotImplementedError
 
     def fit_transform(self, data: TimeSeriesDataFrame) -> TimeSeriesDataFrame:
@@ -45,7 +45,7 @@ class LocalTargetScaler(TargetScaler):
             self.scale = self.scale.clip(lower=self.min_scale).replace([np.inf, -np.inf], np.nan).fillna(1.0)
         return self
 
-    def _reindex_loc_scale(self, item_index: pd.Index) -> Tuple[Union[np.ndarray, float], Union[np.ndarray, float]]:
+    def _reindex_loc_scale(self, item_index: pd.Index) -> tuple[Union[np.ndarray, float], Union[np.ndarray, float]]:
         """Reindex loc and scale parameters for the given item_ids and convert them to an array-like."""
         if self.loc is not None:
             loc = self.loc.reindex(item_index).to_numpy()
@@ -74,7 +74,7 @@ class LocalStandardScaler(LocalTargetScaler):
     The resulting affine transformation is (x - loc) / scale, where scale = std(x), loc = mean(x).
     """
 
-    def _compute_loc_scale(self, target_series: pd.Series) -> Tuple[pd.Series, pd.Series]:
+    def _compute_loc_scale(self, target_series: pd.Series) -> tuple[pd.Series, pd.Series]:
         stats = target_series.groupby(level=ITEMID, sort=False).agg(["mean", "std"])
         return stats["mean"], stats["std"]
 
@@ -82,7 +82,7 @@ class LocalStandardScaler(LocalTargetScaler):
 class LocalMeanAbsScaler(LocalTargetScaler):
     """Applies mean absolute scaling to each time series in the dataset."""
 
-    def _compute_loc_scale(self, target_series: pd.Series) -> Tuple[Optional[pd.Series], pd.Series]:
+    def _compute_loc_scale(self, target_series: pd.Series) -> tuple[Optional[pd.Series], pd.Series]:
         scale = target_series.abs().groupby(level=ITEMID, sort=False).agg("mean")
         return None, scale
 
@@ -93,7 +93,7 @@ class LocalMinMaxScaler(LocalTargetScaler):
     The resulting affine transformation is (x - loc) / scale, where scale = max(x) - min(x), loc = min(x) / scale.
     """
 
-    def _compute_loc_scale(self, target_series: pd.Series) -> Tuple[pd.Series, pd.Series]:
+    def _compute_loc_scale(self, target_series: pd.Series) -> tuple[pd.Series, pd.Series]:
         stats = target_series.abs().groupby(level=ITEMID, sort=False).agg(["min", "max"])
         scale = (stats["max"] - stats["min"]).clip(lower=self.min_scale)
         loc = stats["min"]
@@ -117,7 +117,7 @@ class LocalRobustScaler(LocalTargetScaler):
         self.q_max = 0.75
         assert 0 < self.q_min < self.q_max < 1
 
-    def _compute_loc_scale(self, target_series: pd.Series) -> Tuple[pd.Series, pd.Series]:
+    def _compute_loc_scale(self, target_series: pd.Series) -> tuple[pd.Series, pd.Series]:
         grouped = target_series.groupby(level=ITEMID, sort=False)
         loc = grouped.median()
         lower = grouped.quantile(self.q_min)

autogluon/timeseries/utils/datetime/lags.py

@@ -2,7 +2,7 @@
 Generate lag indices based on frequency string. Adapted from gluonts.time_feature.lag.
 """
 
-from typing import List, Optional
+from typing import Optional
 
 import numpy as np
 import pandas as pd
@@ -72,7 +72,7 @@ def get_lags_for_frequency(
     lag_ub: int = 1200,
     num_lags: Optional[int] = None,
     num_default_lags: int = 7,
-) -> List[int]:
+) -> list[int]:
     """
     Generates a list of lags that that are appropriate for the given frequency
     string.

autogluon/timeseries/utils/datetime/time_features.py

@@ -2,7 +2,7 @@
 Generate time features based on frequency string. Adapted from gluonts.time_feature.time_feature.
 """
 
-from typing import Callable, List
+from typing import Callable
 
 import numpy as np
 import pandas as pd
@@ -56,7 +56,7 @@ def second_of_minute(index: pd.DatetimeIndex) -> np.ndarray:
     return _normalize(index.second, num=60)
 
 
-def get_time_features_for_frequency(freq) -> List[Callable]:
+def get_time_features_for_frequency(freq) -> list[Callable]:
     features_by_offset_name = {
         "YE": [],
         "QE": [quarter_of_year],

autogluon/timeseries/utils/features.py

@@ -2,7 +2,7 @@ import logging
 import reprlib
 import time
 from dataclasses import asdict, dataclass, field
-from typing import Any, Dict, List, Literal, Optional, Tuple
+from typing import Any, Literal, Optional
 
 import numpy as np
 import pandas as pd
@@ -24,50 +24,50 @@ logger = logging.getLogger(__name__)
 class CovariateMetadata:
     """Provides mapping from different covariate types to columns in the dataset."""
 
-    static_features_cat: List[str] = field(default_factory=list)
-    static_features_real: List[str] = field(default_factory=list)
-    known_covariates_real: List[str] = field(default_factory=list)
-    known_covariates_cat: List[str] = field(default_factory=list)
-    past_covariates_real: List[str] = field(default_factory=list)
-    past_covariates_cat: List[str] = field(default_factory=list)
+    static_features_cat: list[str] = field(default_factory=list)
+    static_features_real: list[str] = field(default_factory=list)
+    known_covariates_real: list[str] = field(default_factory=list)
+    known_covariates_cat: list[str] = field(default_factory=list)
+    past_covariates_real: list[str] = field(default_factory=list)
+    past_covariates_cat: list[str] = field(default_factory=list)
 
     @property
-    def static_features(self) -> List[str]:
+    def static_features(self) -> list[str]:
         return self.static_features_cat + self.static_features_real
 
     @property
-    def known_covariates(self) -> List[str]:
+    def known_covariates(self) -> list[str]:
         return self.known_covariates_cat + self.known_covariates_real
 
     @property
-    def past_covariates(self) -> List[str]:
+    def past_covariates(self) -> list[str]:
         return self.past_covariates_cat + self.past_covariates_real
 
     @property
-    def covariates(self) -> List[str]:
+    def covariates(self) -> list[str]:
         return self.known_covariates + self.past_covariates
 
     @property
-    def covariates_real(self) -> List[str]:
+    def covariates_real(self) -> list[str]:
         return self.known_covariates_real + self.past_covariates_real
 
     @property
-    def covariates_cat(self) -> List[str]:
+    def covariates_cat(self) -> list[str]:
         return self.known_covariates_cat + self.past_covariates_cat
 
     @property
-    def real_features(self) -> List[str]:
+    def real_features(self) -> list[str]:
         return self.static_features_real + self.covariates_real
 
     @property
-    def cat_features(self) -> List[str]:
+    def cat_features(self) -> list[str]:
         return self.static_features_cat + self.covariates_cat
 
     @property
-    def all_features(self) -> List[str]:
+    def all_features(self) -> list[str]:
         return self.static_features + self.covariates
 
-    def to_dict(self) -> Dict[str, Any]:
+    def to_dict(self) -> dict[str, Any]:
         return asdict(self)
 
 
@@ -120,13 +120,13 @@ class TimeSeriesFeatureGenerator:
 
     Parameters
     ----------
-    target : str
+    target
         Name of the target column.
-    known_covariates_names : List[str]
+    known_covariates_names
         Columns that contain covariates that are known into the future.
-    float_dtype : str, default = "float32"
+    float_dtype
         Numpy float dtype to which all numeric columns (float, int, bool) will be converted both in static & dynamic dfs.
-    num_samples : int or None, default = 20_000
+    num_samples
         Number of rows sampled from the training dataset to speed up computation of the median (used later for imputation).
         If set to `None`, median will be computed using all rows.
     """
@@ -134,7 +134,7 @@ class TimeSeriesFeatureGenerator:
     def __init__(
         self,
         target: str,
-        known_covariates_names: List[str],
+        known_covariates_names: list[str],
         float_dtype: str = "float32",
         num_samples: Optional[int] = 20_000,
     ):
@@ -143,8 +143,8 @@ class TimeSeriesFeatureGenerator:
         self.num_samples = num_samples
 
         self._is_fit = False
-        self.known_covariates_names: List[str] = list(known_covariates_names)
-        self.past_covariates_names: List[str] = []
+        self.known_covariates_names: list[str] = list(known_covariates_names)
+        self.past_covariates_names: list[str] = []
         self.known_covariates_pipeline = ContinuousAndCategoricalFeatureGenerator()
         self.past_covariates_pipeline = ContinuousAndCategoricalFeatureGenerator()
         # Cat features with cat_count=1 are fine in static_features since they are repeated for all time steps in a TS
@@ -154,7 +154,7 @@ class TimeSeriesFeatureGenerator:
         self._train_static_real_median: Optional[pd.Series] = None
 
     @property
-    def required_column_names(self) -> List[str]:
+    def required_column_names(self) -> list[str]:
         return [self.target] + list(self.known_covariates_names) + list(self.past_covariates_names)
 
     @property
@@ -262,13 +262,13 @@ class TimeSeriesFeatureGenerator:
         return self._impute_covariates(ts_df, column_names=self.covariate_metadata.covariates_real)
 
     @staticmethod
-    def _concat_dfs(dfs_to_concat: List[pd.DataFrame]) -> pd.DataFrame:
+    def _concat_dfs(dfs_to_concat: list[pd.DataFrame]) -> pd.DataFrame:
         if len(dfs_to_concat) == 1:
             return dfs_to_concat[0]
         else:
             return pd.concat(dfs_to_concat, axis=1, copy=False)
 
-    def _impute_covariates(self, ts_df: TimeSeriesDataFrame, column_names: List[str]) -> TimeSeriesDataFrame:
+    def _impute_covariates(self, ts_df: TimeSeriesDataFrame, column_names: list[str]) -> TimeSeriesDataFrame:
         """Impute missing values in selected columns with ffill, bfill, and median imputation."""
         if len(column_names) > 0:
             # ffill + bfill covariates that have at least some observed values
@@ -346,10 +346,10 @@ class TimeSeriesFeatureGenerator:
             return None
 
     @staticmethod
-    def _detect_and_log_column_types(transformed_df: pd.DataFrame) -> Tuple[List[str], List[str]]:
+    def _detect_and_log_column_types(transformed_df: pd.DataFrame) -> tuple[list[str], list[str]]:
         """Log & return names of categorical and real-valued columns in the DataFrame."""
-        cat_column_names: List[str] = []
-        real_column_names: List[str] = []
+        cat_column_names: list[str] = []
+        real_column_names: list[str] = []
         for column_name, column_dtype in transformed_df.dtypes.items():
             if isinstance(column_dtype, pd.CategoricalDtype):
                 cat_column_names.append(str(column_name))
@@ -362,9 +362,9 @@ class TimeSeriesFeatureGenerator:
 
     @staticmethod
     def _check_required_columns_are_present(
-        data: TimeSeriesDataFrame, required_column_names: List[str], data_frame_name: str
+        data: TimeSeriesDataFrame, required_column_names: list[str], data_frame_name: str
     ) -> None:
-        missing_columns = pd.Index(required_column_names).difference(data.columns)
+        missing_columns = pd.Index(required_column_names).difference(data.columns)  # type: ignore
         if len(missing_columns) > 0:
             raise ValueError(
                 f"{len(missing_columns)} columns are missing from {data_frame_name}: {reprlib.repr(missing_columns.to_list())}"

autogluon/timeseries/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.4.1b20250820"
+__version__ = "1.4.1b20250902"
 __lite__ = False

{autogluon.timeseries-1.4.1b20250820.dist-info → autogluon.timeseries-1.4.1b20250902.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 1.4.1b20250820
+Version: 1.4.1b20250902
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -55,10 +55,10 @@ Requires-Dist: fugue>=0.9.0
 Requires-Dist: tqdm<5,>=4.38
 Requires-Dist: orjson~=3.9
 Requires-Dist: tensorboard<3,>=2.9
-Requires-Dist: autogluon.core[raytune]==1.4.1b20250820
-Requires-Dist: autogluon.common==1.4.1b20250820
-Requires-Dist: autogluon.features==1.4.1b20250820
-Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.4.1b20250820
+Requires-Dist: autogluon.core[raytune]==1.4.1b20250902
+Requires-Dist: autogluon.common==1.4.1b20250902
+Requires-Dist: autogluon.features==1.4.1b20250902
+Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.4.1b20250902
 Provides-Extra: all
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"

autogluon.timeseries-1.4.1b20250902.dist-info/RECORD

@@ -0,0 +1,75 @@
+autogluon.timeseries-1.4.1b20250902-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon/timeseries/__init__.py,sha256=_CrLLc1fkjen7UzWoO0Os8WZoHOgvZbHKy46I8v_4k4,304
+autogluon/timeseries/evaluator.py,sha256=l642tYfTHsl8WVIq_vV6qhgAFVFr9UuZD7gLra3A_Kc,250
+autogluon/timeseries/learner.py,sha256=eQrqFVOmL-2JC85LgCMkbyoLpKS02Dilg1T8RUeS_LI,13887
+autogluon/timeseries/predictor.py,sha256=7X4YsWYa3Xk2RI1Irf2O-c3-I82Zqhg-cgj8cj_4AoA,88427
+autogluon/timeseries/regressor.py,sha256=lc8Qr3-8v4oxajtCnV3sxpUaW6vxXXJOA6Kr-qVne4k,11926
+autogluon/timeseries/splitter.py,sha256=8ACkuCXeUhQGUx4jz_Vv17q814WrHJQeKvq2v4-oE6s,3158
+autogluon/timeseries/version.py,sha256=HXEe1UreeuIzhlJpXoFLQx29wpNs2WiCSUMbk708j5k,91
+autogluon/timeseries/configs/__init__.py,sha256=wiLBwxZkDTQBJkSJ9-xz3p_yJxX0dbHe108dS1P5O6A,183
+autogluon/timeseries/configs/hyperparameter_presets.py,sha256=GbI2sd3uakWtaeaMyF7B5z_lmyfb6ToK6PZEUZTyG9w,2031
+autogluon/timeseries/configs/predictor_presets.py,sha256=B5HFHIelh91hhG0YYE5SJ7_14P7sylFAABgHX8n_53M,2712
+autogluon/timeseries/dataset/__init__.py,sha256=UvnhAN5tjgxXTHoZMQDy64YMDj4Xxa68yY7NP4vAw0o,81
+autogluon/timeseries/dataset/ts_dataframe.py,sha256=EwxKBScspwKnJTqIk2Icukk8vIrbKYObOMAkNIn4zc8,51760
+autogluon/timeseries/metrics/__init__.py,sha256=YJPXxsJ0tRDXq7p-sTZSLb0DuXMJH6sT1PgbZ3tMt30,3594
+autogluon/timeseries/metrics/abstract.py,sha256=6jbluvHXfLc_cuK1Fx0ZYle2sR4WGG6YxFQhkor46Q8,11545
+autogluon/timeseries/metrics/point.py,sha256=sS__n_Em7m4CUaBu3PNWQ_dHw1YCOHbEyC15fhytFL8,18308
+autogluon/timeseries/metrics/quantile.py,sha256=x0cq44fXRoMiuI4BVQ7mpWk1YgrK4OwLTlJAhCHQ7Xg,4634
+autogluon/timeseries/metrics/utils.py,sha256=HuDe1BNe8yJU4f_DKM913nNrUueoRaw6zhxm1-S20s0,910
+autogluon/timeseries/models/__init__.py,sha256=9YnqkOILtVEkbICk7J3VlMkMNySs-f5ErIUKrE5-fys,1294
+autogluon/timeseries/models/registry.py,sha256=8n7W04ql0ckNQUzKcAW7bxreLI8wTAUTymACgLklH9M,2158
+autogluon/timeseries/models/abstract/__init__.py,sha256=Htfkjjc3vo92RvyM8rIlQ0PLWt3jcrCKZES07UvCMV0,146
+autogluon/timeseries/models/abstract/abstract_timeseries_model.py,sha256=97HOi7fRPxtx8Y9hq-xdJI-kLMp6Z-8LUSvcfBjXFsM,31978
+autogluon/timeseries/models/abstract/model_trial.py,sha256=ENPg_7nsdxIvaNM0o0UShZ3x8jFlRmwRc5m0fGPC0TM,3720
+autogluon/timeseries/models/abstract/tunable.py,sha256=jA6p-FPZkMva67B-1foqvHK-1rr0IdEfp9RvGW1WS9I,7155
+autogluon/timeseries/models/autogluon_tabular/__init__.py,sha256=E5fZsdFPgVdyCVyj5bGmn_lQFlCMn2NvuRLBMcCFvhM,205
+autogluon/timeseries/models/autogluon_tabular/mlforecast.py,sha256=k3a0JqBeuLQfjCtZ8MA7UvS2eqHjwbw0-4kN_StMMUQ,37623
+autogluon/timeseries/models/autogluon_tabular/per_step.py,sha256=M5rhj_jjcQz27wPYm6NEBEE0aHgXe0Bl6HFc2NIuLdU,23159
+autogluon/timeseries/models/autogluon_tabular/transforms.py,sha256=aI1QJLJaOB5Xy2WA0jo6Jh25MRVyyZ8ONrqlV96kpw0,2735
+autogluon/timeseries/models/autogluon_tabular/utils.py,sha256=Fn3Vu_Q0PCtEUbtNgLp1xIblg7dOdpFlF3W5kLHgruI,63
+autogluon/timeseries/models/chronos/__init__.py,sha256=wT77HzTtmQxW3sw2k0mA5Ot6PSHivX-Uvn5fjM05EU4,60
+autogluon/timeseries/models/chronos/model.py,sha256=UYLI1HVwsW5KfA-jXqJdBel-7N6l6ggdBVrjLLaq9P0,32333
+autogluon/timeseries/models/chronos/pipeline/__init__.py,sha256=bkTR0LSKIxAaKFOr9A0HSkCtnRdikDPUPp810WOKgxE,247
+autogluon/timeseries/models/chronos/pipeline/base.py,sha256=Us-TUpHSN3mM3ut05IVc2a9Q6KYq1n9pTb7JZG7b6kA,5546
+autogluon/timeseries/models/chronos/pipeline/chronos.py,sha256=bgow5FkHG7y5qWBXcggqXemnistJUfrl0lWFXcGXg5g,20197
+autogluon/timeseries/models/chronos/pipeline/chronos_bolt.py,sha256=KJYgxASTW2VhS0ObkP5DUQXyfnTRTXzjRD5Gm-FQFI4,21355
+autogluon/timeseries/models/chronos/pipeline/utils.py,sha256=WYeCKFP5dxs4u09XTncBI2486VV22O1DiM9a3ZvZ1OE,12790
+autogluon/timeseries/models/ensemble/__init__.py,sha256=x2Y6dWk15XugTEWNUKq8U5z6nIjelo3UjpI-TfS13OE,159
+autogluon/timeseries/models/ensemble/abstract.py,sha256=wvtXNZTwiYpIurPkOYSzsi3XTRRx5guJLMYLmXTdOeQ,5695
+autogluon/timeseries/models/ensemble/basic.py,sha256=aSQRYylUpFZVk_Lpv5GY8uYgmE0_ipLy_tx6ELTZyWc,3426
+autogluon/timeseries/models/ensemble/greedy.py,sha256=zXJFenn1XxNNvCp4TlmIq1Dx3pUDWjKG1K3HsejmDeY,7323
+autogluon/timeseries/models/gluonts/__init__.py,sha256=YfyNYOkhhNsloA4MAavfmqKO29_q6o4lwPoV7L4_h7M,355
+autogluon/timeseries/models/gluonts/abstract.py,sha256=fYXV5fQk79LtPtpa4uk8MzCUGZ5J6n47ClEYuYSBDLY,27770
+autogluon/timeseries/models/gluonts/dataset.py,sha256=iUVKZyec1efVW2-71AWn-m9cowXYYj5hJM1JWuioscA,5115
+autogluon/timeseries/models/gluonts/models.py,sha256=1Z3x3-jVoae5X4cSnDIgJMvTJ9_O94aDSW8HEnBaL5k,25907
+autogluon/timeseries/models/local/__init__.py,sha256=e2UImoJhmj70E148IIObv90C_bHxgyLNk6YsS4p7pfs,701
+autogluon/timeseries/models/local/abstract_local_model.py,sha256=A3sNYMA67UbbEKIIN30BcBdE_NpwaBtcG22O5mVWS6k,11482
+autogluon/timeseries/models/local/naive.py,sha256=xur3WWhLaS9Iix_p_yfaStbr58nL5K4rV0dReTm3BQQ,7496
+autogluon/timeseries/models/local/npts.py,sha256=VRZk5tEJOIentt0tLM6lxyoU8US736nHOvhSAgagYMc,4203
+autogluon/timeseries/models/local/statsforecast.py,sha256=sZ6aEFzAyPNZX3rMULGWFht0Toapjb3EwHe5Rb76ZxA,33318
+autogluon/timeseries/models/multi_window/__init__.py,sha256=Bq7AT2Jxdd4WNqmjTdzeqgNiwn1NCyWp4tBIWaM-zfI,60
+autogluon/timeseries/models/multi_window/multi_window_model.py,sha256=Hn-H2jLdeuB0_TxhAdununS8ti-iO-WSl3FOoxzcEJA,12369
+autogluon/timeseries/trainer/__init__.py,sha256=_tw3iioJfvtIV7wnjtEMv0yS8oabmCFxDnGRodYE7RI,72
+autogluon/timeseries/trainer/model_set_builder.py,sha256=s6tozfND3lLfst6Vxa_oP_wgCmDapyCJYFmCjkEn-es,10788
+autogluon/timeseries/trainer/prediction_cache.py,sha256=Vi6EbMiMheq_smA93U_MoMxYUV85RdPm0dvJFdsM8K4,5551
+autogluon/timeseries/trainer/trainer.py,sha256=LF2X5UNnrU8w5h_i09SphGWvGFvZ6KvPDq89Z3GzZZQ,54959
+autogluon/timeseries/transforms/__init__.py,sha256=fKlT4pkJ_8Gl7IUTc3uSDzt2Xow5iH5w6fPB3ePNrTg,127
+autogluon/timeseries/transforms/covariate_scaler.py,sha256=9lEfDS4wnVZohQNnm9OcAXr3voUl83RCnctKR3O66iU,7030
+autogluon/timeseries/transforms/target_scaler.py,sha256=kTQrXAsDHCnYuqfpaVuvefyTgyp_ylDpUIPz7pArjeY,6043
+autogluon/timeseries/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autogluon/timeseries/utils/features.py,sha256=tdL7jZKeySO7dgB09FweR44wPCmfWg8-ZM5uVzeyvYQ,22593
+autogluon/timeseries/utils/forecast.py,sha256=yK1_eNtRUPYGs0R-VWMO4c81LrTGF57ih3yzsXVHyGY,2191
+autogluon/timeseries/utils/warning_filters.py,sha256=SroNhLU3kwbD8anM58vdxWq36Z8j_uiY42mEt0ya-JI,2589
+autogluon/timeseries/utils/datetime/__init__.py,sha256=bTMR8jLh1LW55vHjbOr1zvWRMF_PqbvxpS-cUcNIDWI,173
+autogluon/timeseries/utils/datetime/base.py,sha256=3NdsH3NDq4cVAOSoy3XpaNixyNlbjy4DJ_YYOGuu9x4,1341
+autogluon/timeseries/utils/datetime/lags.py,sha256=rjJtdBU0M41R1jwfmvCbo045s-6XBjhGVnGBQJ9-U1E,5997
+autogluon/timeseries/utils/datetime/seasonality.py,sha256=YK_2k8hvYIMW-sJPnjGWRtCnvIOthwA2hATB3nwVoD4,834
+autogluon/timeseries/utils/datetime/time_features.py,sha256=kEOFls4Nzh8nO0Pcz1DwLsC_NA3hMI4JUlZI3kuvuts,2666
+autogluon.timeseries-1.4.1b20250902.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.timeseries-1.4.1b20250902.dist-info/METADATA,sha256=kLL8EOiGm0gBy69gkuclLiOKWuRa8mrYGAW1JjX5gec,12463
+autogluon.timeseries-1.4.1b20250902.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.timeseries-1.4.1b20250902.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.timeseries-1.4.1b20250902.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-1.4.1b20250902.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-1.4.1b20250902.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.timeseries-1.4.1b20250902.dist-info/RECORD,,

autogluon/timeseries/configs/presets_configs.py

@@ -1,79 +0,0 @@
-"""Preset configurations for autogluon.timeseries Predictors"""
-
-from autogluon.timeseries.models.presets import get_default_hps
-
-# TODO: change default HPO settings when other HPO strategies (e.g., Ray tune) are available
-# TODO: add refit_full arguments once refitting is available
-
-TIMESERIES_PRESETS_CONFIGS = dict(
-    best_quality={"hyperparameters": "default", "num_val_windows": 2},
-    high_quality={"hyperparameters": "default"},
-    medium_quality={"hyperparameters": "light"},
-    fast_training={"hyperparameters": "very_light"},
-    # Chronos-Bolt models
-    bolt_tiny={
-        "hyperparameters": {"Chronos": {"model_path": "bolt_tiny"}},
-        "skip_model_selection": True,
-    },
-    bolt_mini={
-        "hyperparameters": {"Chronos": {"model_path": "bolt_mini"}},
-        "skip_model_selection": True,
-    },
-    bolt_small={
-        "hyperparameters": {"Chronos": {"model_path": "bolt_small"}},
-        "skip_model_selection": True,
-    },
-    bolt_base={
-        "hyperparameters": {"Chronos": {"model_path": "bolt_base"}},
-        "skip_model_selection": True,
-    },
-    # Original Chronos models
-    chronos_tiny={
-        "hyperparameters": {"Chronos": {"model_path": "tiny"}},
-        "skip_model_selection": True,
-    },
-    chronos_mini={
-        "hyperparameters": {"Chronos": {"model_path": "mini"}},
-        "skip_model_selection": True,
-    },
-    chronos_small={
-        "hyperparameters": {"Chronos": {"model_path": "small"}},
-        "skip_model_selection": True,
-    },
-    chronos_base={
-        "hyperparameters": {"Chronos": {"model_path": "base"}},
-        "skip_model_selection": True,
-    },
-    chronos_large={
-        "hyperparameters": {"Chronos": {"model_path": "large", "batch_size": 8}},
-        "skip_model_selection": True,
-    },
-    chronos_ensemble={
-        "hyperparameters": {
-            "Chronos": {"model_path": "small"},
-            **get_default_hps("light_inference"),
-        }
-    },
-    chronos_large_ensemble={
-        "hyperparameters": {
-            "Chronos": {"model_path": "large", "batch_size": 8},
-            **get_default_hps("light_inference"),
-        }
-    },
-)
-
-TIMESERIES_PRESETS_ALIASES = dict(
-    chronos="chronos_small",
-    best="best_quality",
-    high="high_quality",
-    medium="medium_quality",
-    bq="best_quality",
-    hq="high_quality",
-    mq="medium_quality",
-)
-
-# update with aliases
-TIMESERIES_PRESETS_CONFIGS = {
-    **TIMESERIES_PRESETS_CONFIGS,
-    **{k: TIMESERIES_PRESETS_CONFIGS[v].copy() for k, v in TIMESERIES_PRESETS_ALIASES.items()},
-}