autogluon.timeseries 1.3.2b20250712__py3-none-any.whl → 1.4.1b20251116__py3-none-any.whl

autogluon/timeseries/trainer/utils.py

@@ -0,0 +1,18 @@
+import logging
+from typing import Optional
+
+logger = logging.getLogger("autogluon.timeseries.trainer")
+
+
+def log_scores_and_times(
+    val_score: Optional[float],
+    fit_time: Optional[float],
+    predict_time: Optional[float],
+    eval_metric_name: str,
+):
+    if val_score is not None:
+        logger.info(f"\t{val_score:<7.4f}".ljust(15) + f"= Validation score ({eval_metric_name})")
+    if fit_time is not None:
+        logger.info(f"\t{fit_time:<7.2f} s".ljust(15) + "= Training runtime")
+    if predict_time is not None:
+        logger.info(f"\t{predict_time:<7.2f} s".ljust(15) + "= Validation (prediction) runtime")

autogluon/timeseries/transforms/covariate_scaler.py

@@ -1,12 +1,12 @@
 import logging
-from typing import Dict, List, Literal, Optional, Protocol, overload, runtime_checkable
+from typing import Literal, Optional, Protocol, overload, runtime_checkable
 
 import numpy as np
 import pandas as pd
 from sklearn.compose import ColumnTransformer
 from sklearn.preprocessing import QuantileTransformer, StandardScaler
 
-from autogluon.timeseries.dataset.ts_dataframe import TimeSeriesDataFrame
+from autogluon.timeseries.dataset import TimeSeriesDataFrame
 from autogluon.timeseries.utils.features import CovariateMetadata
 from autogluon.timeseries.utils.warning_filters import warning_filter
 
@@ -53,7 +53,7 @@ class GlobalCovariateScaler(CovariateScaler):
         self.use_past_covariates = use_past_covariates
         self.use_static_features = use_static_features
         self.skew_threshold = skew_threshold
-        self._column_transformers: Optional[Dict[Literal["known", "past", "static"], ColumnTransformer]] = None
+        self._column_transformers: Optional[dict[Literal["known", "past", "static"], ColumnTransformer]] = None
 
     def is_fit(self) -> bool:
         return self._column_transformers is not None
@@ -117,7 +117,7 @@ class GlobalCovariateScaler(CovariateScaler):
             known_covariates[columns] = self._column_transformers["known"].transform(known_covariates[columns])
         return known_covariates
 
-    def _get_transformer_for_columns(self, df: pd.DataFrame, columns: List[str]) -> ColumnTransformer:
+    def _get_transformer_for_columns(self, df: pd.DataFrame, columns: list[str]) -> ColumnTransformer:
         """Passthrough bool features, use QuantileTransform for skewed features, and use StandardScaler for the rest.
 
         The preprocessing logic is similar to the TORCH_NN model from Tabular.

autogluon/timeseries/transforms/target_scaler.py

@@ -1,10 +1,10 @@
-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
 from typing_extensions import Self
 
-from autogluon.timeseries.dataset.ts_dataframe import ITEMID, TimeSeriesDataFrame
+from autogluon.timeseries.dataset import TimeSeriesDataFrame
 
 
 class TargetScaler(Protocol):
@@ -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()
@@ -59,12 +59,12 @@ class LocalTargetScaler(TargetScaler):
 
     def transform(self, data: TimeSeriesDataFrame) -> TimeSeriesDataFrame:
         """Apply scaling to the target column in the dataframe."""
-        loc, scale = self._reindex_loc_scale(item_index=data.index.get_level_values(ITEMID))
+        loc, scale = self._reindex_loc_scale(item_index=data.index.get_level_values(TimeSeriesDataFrame.ITEMID))
         return data.assign(**{self.target: (data[self.target] - loc) / scale})
 
     def inverse_transform(self, predictions: TimeSeriesDataFrame) -> TimeSeriesDataFrame:
         """Apply inverse scaling to all columns in the predictions dataframe."""
-        loc, scale = self._reindex_loc_scale(item_index=predictions.index.get_level_values(ITEMID))
+        loc, scale = self._reindex_loc_scale(item_index=predictions.index.get_level_values(TimeSeriesDataFrame.ITEMID))
         return predictions.assign(**{col: predictions[col] * scale + loc for col in predictions.columns})
 
 
@@ -74,16 +74,16 @@ 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]:
-        stats = target_series.groupby(level=ITEMID, sort=False).agg(["mean", "std"])
+    def _compute_loc_scale(self, target_series: pd.Series) -> tuple[pd.Series, pd.Series]:
+        stats = target_series.groupby(level=TimeSeriesDataFrame.ITEMID, sort=False).agg(["mean", "std"])
         return stats["mean"], stats["std"]
 
 
 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]:
-        scale = target_series.abs().groupby(level=ITEMID, sort=False).agg("mean")
+    def _compute_loc_scale(self, target_series: pd.Series) -> tuple[Optional[pd.Series], pd.Series]:
+        scale = target_series.abs().groupby(level=TimeSeriesDataFrame.ITEMID, sort=False).agg("mean")
         return None, scale
 
 
@@ -93,8 +93,8 @@ 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]:
-        stats = target_series.abs().groupby(level=ITEMID, sort=False).agg(["min", "max"])
+    def _compute_loc_scale(self, target_series: pd.Series) -> tuple[pd.Series, pd.Series]:
+        stats = target_series.abs().groupby(level=TimeSeriesDataFrame.ITEMID, sort=False).agg(["min", "max"])
         scale = (stats["max"] - stats["min"]).clip(lower=self.min_scale)
         loc = stats["min"]
         return loc, scale
@@ -117,8 +117,8 @@ 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]:
-        grouped = target_series.groupby(level=ITEMID, sort=False)
+    def _compute_loc_scale(self, target_series: pd.Series) -> tuple[pd.Series, pd.Series]:
+        grouped = target_series.groupby(level=TimeSeriesDataFrame.ITEMID, sort=False)
         loc = grouped.median()
         lower = grouped.quantile(self.q_min)
         upper = grouped.quantile(self.q_max)

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
@@ -14,7 +14,7 @@ from autogluon.features.generators import (
     IdentityFeatureGenerator,
     PipelineFeatureGenerator,
 )
-from autogluon.timeseries.dataset.ts_dataframe import ITEMID, TimeSeriesDataFrame
+from autogluon.timeseries.dataset import TimeSeriesDataFrame
 from autogluon.timeseries.utils.warning_filters import warning_filter
 
 logger = logging.getLogger(__name__)
@@ -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())}"
@@ -415,7 +415,9 @@ class AbstractFeatureImportanceTransform:
         if feature_name in self.covariate_metadata.past_covariates:
             # we'll have to work on the history of the data alone
             data[feature_name] = data[feature_name].copy()
-            feature_data = data[feature_name].groupby(level=ITEMID, sort=False).head(-self.prediction_length)
+            feature_data = (
+                data[feature_name].groupby(level=TimeSeriesDataFrame.ITEMID, sort=False).head(-self.prediction_length)
+            )
             # Silence spurious FutureWarning raised by DataFrame.update https://github.com/pandas-dev/pandas/issues/57124
             with warning_filter():
                 data[feature_name].update(self._transform_series(feature_data, is_categorical=is_categorical))
@@ -455,7 +457,7 @@ class PermutationFeatureImportanceTransform(AbstractFeatureImportanceTransform):
         rng = np.random.RandomState(self.random_seed)
 
         if self.shuffle_type == "itemwise":
-            return feature_data.groupby(level=ITEMID, sort=False).transform(
+            return feature_data.groupby(level=TimeSeriesDataFrame.ITEMID, sort=False).transform(
                 lambda x: x.sample(frac=1, random_state=rng).values
             )
         elif self.shuffle_type == "naive":
@@ -483,6 +485,8 @@ class ConstantReplacementFeatureImportanceTransform(AbstractFeatureImportanceTra
 
     def _transform_series(self, feature_data: pd.Series, is_categorical: bool) -> pd.Series:
         if is_categorical:
-            return feature_data.groupby(level=ITEMID, sort=False).transform(lambda x: x.mode()[0])
+            return feature_data.groupby(level=TimeSeriesDataFrame.ITEMID, sort=False).transform(lambda x: x.mode()[0])
         else:
-            return feature_data.groupby(level=ITEMID, sort=False).transform(self.real_value_aggregation)
+            return feature_data.groupby(level=TimeSeriesDataFrame.ITEMID, sort=False).transform(
+                self.real_value_aggregation
+            )  # type: ignore

autogluon/timeseries/utils/forecast.py

@@ -5,7 +5,7 @@ import numpy as np
 import pandas as pd
 
 from autogluon.common.utils.deprecated_utils import Deprecated
-from autogluon.timeseries.dataset.ts_dataframe import ITEMID, TIMESTAMP, TimeSeriesDataFrame
+from autogluon.timeseries.dataset import TimeSeriesDataFrame
 
 
 def get_forecast_horizon_index_single_time_series(
@@ -16,7 +16,7 @@ def get_forecast_horizon_index_single_time_series(
     if offset is None:
         raise ValueError(f"Invalid frequency: {freq}")
     start_ts = past_timestamps.max() + 1 * offset
-    return pd.date_range(start=start_ts, periods=prediction_length, freq=freq, name=TIMESTAMP)
+    return pd.date_range(start=start_ts, periods=prediction_length, freq=freq, name=TimeSeriesDataFrame.TIMESTAMP)
 
 
 @Deprecated(
@@ -37,14 +37,14 @@ def make_future_data_frame(
     """
     indptr = ts_dataframe.get_indptr()
     last = ts_dataframe.index[indptr[1:] - 1].to_frame(index=False)
-    item_ids = np.repeat(last[ITEMID].to_numpy(), prediction_length)
+    item_ids = np.repeat(last[TimeSeriesDataFrame.ITEMID].to_numpy(), prediction_length)
 
     if freq is None:
         freq = ts_dataframe.freq
     offset = pd.tseries.frequencies.to_offset(freq)
-    last_ts = pd.DatetimeIndex(last[TIMESTAMP])
+    last_ts = pd.DatetimeIndex(last[TimeSeriesDataFrame.TIMESTAMP])
     # Non-vectorized offsets like BusinessDay may produce a PerformanceWarning - we filter them
     with warnings.catch_warnings():
         warnings.simplefilter("ignore", category=pd.errors.PerformanceWarning)
         timestamps = np.dstack([last_ts + step * offset for step in range(1, prediction_length + 1)]).ravel()  # type: ignore[operator]
-    return pd.DataFrame({ITEMID: item_ids, TIMESTAMP: timestamps})
+    return pd.DataFrame({TimeSeriesDataFrame.ITEMID: item_ids, TimeSeriesDataFrame.TIMESTAMP: timestamps})

autogluon/timeseries/utils/warning_filters.py

@@ -8,12 +8,14 @@ import sys
 import warnings
 from collections import Counter
 
+import pandas as pd
+
 __all__ = ["warning_filter", "disable_root_logger", "disable_tqdm"]
 
 
 @contextlib.contextmanager
 def warning_filter(all_warnings: bool = False):
-    categories = [RuntimeWarning, UserWarning, FutureWarning]
+    categories = [RuntimeWarning, UserWarning, FutureWarning, pd.errors.PerformanceWarning]
     if all_warnings:
         categories.append(Warning)
     with warnings.catch_warnings():

autogluon/timeseries/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.3.2b20250712"
+__version__ = "1.4.1b20251116"
 __lite__ = False

autogluon.timeseries-1.4.1b20251116-py3.9-nspkg.pth

@@ -0,0 +1 @@
+import sys, types, os;p = os.path.join(sys._getframe(1).f_locals['sitedir'], *('autogluon',));importlib = __import__('importlib.util');__import__('importlib.machinery');m = sys.modules.setdefault('autogluon', importlib.util.module_from_spec(importlib.machinery.PathFinder.find_spec('autogluon', [os.path.dirname(p)])));m = m or sys.modules.setdefault('autogluon', types.ModuleType('autogluon'));mp = (m or []) and m.__dict__.setdefault('__path__',[]);(p not in mp) and mp.append(p)

{autogluon.timeseries-1.3.2b20250712.dist-info → autogluon_timeseries-1.4.1b20251116.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: autogluon.timeseries
-Version: 1.3.2b20250712
+Version: 1.4.1b20251116
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -9,7 +9,6 @@ Project-URL: Documentation, https://auto.gluon.ai
 Project-URL: Bug Reports, https://github.com/autogluon/autogluon/issues
 Project-URL: Source, https://github.com/autogluon/autogluon/
 Project-URL: Contribute!, https://github.com/autogluon/autogluon/blob/master/CONTRIBUTING.md
-Platform: UNKNOWN
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Education
 Classifier: Intended Audience :: Developers
@@ -34,15 +33,15 @@ Classifier: Topic :: Scientific/Engineering :: Information Analysis
 Classifier: Topic :: Scientific/Engineering :: Image Recognition
 Requires-Python: >=3.9, <3.13
 Description-Content-Type: text/markdown
-License-File: ../LICENSE
-License-File: ../NOTICE
-Requires-Dist: joblib<2,>=1.1
+License-File: LICENSE
+License-File: NOTICE
+Requires-Dist: joblib<1.7,>=1.2
 Requires-Dist: numpy<2.4.0,>=1.25.0
 Requires-Dist: scipy<1.17,>=1.5.4
 Requires-Dist: pandas<2.4.0,>=2.0.0
-Requires-Dist: torch<2.8,>=2.2
-Requires-Dist: lightning<2.8,>=2.2
-Requires-Dist: pytorch-lightning
+Requires-Dist: torch<2.8,>=2.6
+Requires-Dist: lightning<2.8,>=2.5.1
+Requires-Dist: pytorch_lightning
 Requires-Dist: transformers[sentencepiece]<4.50,>=4.38.0
 Requires-Dist: accelerate<2.0,>=0.34.0
 Requires-Dist: gluonts<0.17,>=0.15.0
@@ -54,17 +53,35 @@ Requires-Dist: coreforecast<0.0.17,>=0.0.12
 Requires-Dist: fugue>=0.9.0
 Requires-Dist: tqdm<5,>=4.38
 Requires-Dist: orjson~=3.9
+Requires-Dist: chronos-forecasting<3,>=2.0.1
 Requires-Dist: tensorboard<3,>=2.9
-Requires-Dist: autogluon.core[raytune]==1.3.2b20250712
-Requires-Dist: autogluon.common==1.3.2b20250712
-Requires-Dist: autogluon.features==1.3.2b20250712
-Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.3.2b20250712
-Provides-Extra: all
+Requires-Dist: autogluon.core[raytune]==1.4.1b20251116
+Requires-Dist: autogluon.common==1.4.1b20251116
+Requires-Dist: autogluon.features==1.4.1b20251116
+Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.4.1b20251116
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"
 Requires-Dist: ruff>=0.0.285; extra == "tests"
 Requires-Dist: flaky<4,>=3.7; extra == "tests"
 Requires-Dist: pytest-timeout<3,>=2.1; extra == "tests"
+Provides-Extra: toto
+Requires-Dist: einops<1,>=0.7; extra == "toto"
+Requires-Dist: rotary-embedding-torch<1,>=0.8; extra == "toto"
+Provides-Extra: all
+Requires-Dist: rotary-embedding-torch<1,>=0.8; extra == "all"
+Requires-Dist: einops<1,>=0.7; extra == "all"
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 
 
@@ -108,7 +125,7 @@ Build accurate end-to-end ML models in just 3 lines of code!
 
 ```python
 from autogluon.tabular import TabularPredictor
-predictor = TabularPredictor(label="class").fit("train.csv")
+predictor = TabularPredictor(label="class").fit("train.csv", presets="best")
 predictions = predictor.predict("test.csv")
 ```
 
@@ -165,5 +182,3 @@ We are actively accepting code contributions to the AutoGluon project. If you ar
 ## :classical_building: License
 
 This library is licensed under the Apache 2.0 License.
-
-

autogluon_timeseries-1.4.1b20251116.dist-info/RECORD

@@ -0,0 +1,96 @@
+autogluon.timeseries-1.4.1b20251116-py3.9-nspkg.pth,sha256=kAlKxjI5mE3Pwwqphu2maN5OBQk8W8ew70e_qbI1c6A,482
+autogluon/timeseries/__init__.py,sha256=_CrLLc1fkjen7UzWoO0Os8WZoHOgvZbHKy46I8v_4k4,304
+autogluon/timeseries/learner.py,sha256=XTQgfZs5ZQf_7mWUz-CNnavewrfNy3ENwtGMRJWwwPQ,13889
+autogluon/timeseries/predictor.py,sha256=khISLnhVxTWMhE0WVCcTgm79K4Q9IuC-jHe01A9w1go,87468
+autogluon/timeseries/regressor.py,sha256=X9ItbQ0e3GyLpKqusjMls5uavqw8w53AH0tXfSFmVno,12049
+autogluon/timeseries/splitter.py,sha256=wK335v7cUAVPbo_9Bok1C6TFg0rB9SH3D031m0vn9-A,2342
+autogluon/timeseries/version.py,sha256=i1AvtCpe0Hg2dM7cUojAmubBtfodplDajXyAsahYUGg,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=49Itgcrjej-x22HYMCXPGD2gjCTRkyHpY2H83aD9U9k,52384
+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=3XLKn01R2roLPZqcyAcxAIy_O89hdr0b4IKHyzRrXYA,4621
+autogluon/timeseries/metrics/utils.py,sha256=_Nz6GLbs91WhqN1PoA53wD4xEEuPIQ0juV5l9rDmkFo,970
+autogluon/timeseries/models/__init__.py,sha256=9NY9mqYaZe_7XB70M6psHARH-Lpkfroj4toUUPO9BmI,1339
+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=3p3ukQxWN4WQHKt3ocmIb_5VlZfHwWJikQYUhSbDbtE,36457
+autogluon/timeseries/models/autogluon_tabular/per_step.py,sha256=keEW7M4SIsu3hC4EFuxcrj5s7QjF9k_7NBARuMXmYgA,23329
+autogluon/timeseries/models/autogluon_tabular/transforms.py,sha256=XtxvaRsnmVF8strfvzEfWO5a_Q8p_wMyxHyglpO1R1c,2886
+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=N6tjC8gSOLcL5eX29JYcOgfxlRATGI2qtTZucCD83t8,33437
+autogluon/timeseries/models/chronos/utils.py,sha256=6y2wphSVYR1ylscSGdb3NvrTU4ZDgbx56Gluxht_j-k,14465
+autogluon/timeseries/models/ensemble/__init__.py,sha256=9fthsA6ozZoTC7A33O0hGhiHAMzcAgG206-b4PIF9Yc,1070
+autogluon/timeseries/models/ensemble/abstract.py,sha256=ePsz2lzmludxq4x_R1jjYgPvxMc0yqVRqHbU1Fq_pvo,4264
+autogluon/timeseries/models/ensemble/array_based/__init__.py,sha256=xCzFHS9YTPsC0LPfhh8mOWzUTYxXGz1RJ15ox0Wgr98,159
+autogluon/timeseries/models/ensemble/array_based/abstract.py,sha256=RC0PL4LvU7REF_FdQwqGT9TmeETNjFlHOJSBTeJrER8,10330
+autogluon/timeseries/models/ensemble/array_based/models.py,sha256=yvqWgXZU2iKxSe4J-kbEYHA3Lah8bYUG2-hdMNMlLP4,1640
+autogluon/timeseries/models/ensemble/array_based/regressor/__init__.py,sha256=Fw5m77f8Z5Y6UrgYFsK7bi1fIgLWdqzvoWZqkfAVmmY,327
+autogluon/timeseries/models/ensemble/array_based/regressor/abstract.py,sha256=cYsmZcjUg84EROimaBUI3X-EPIT4xGyEEqHfHFbiGYQ,2615
+autogluon/timeseries/models/ensemble/array_based/regressor/per_quantile_tabular.py,sha256=oJezyB0Kv2GPChF-Ym9IsyRX4N3OYcUx32hejvMVMTI,5061
+autogluon/timeseries/models/ensemble/array_based/regressor/tabular.py,sha256=2FnOiBDVgaldOnQcPD77mNzXJq7EDb1FGMpwjA3KAlE,4763
+autogluon/timeseries/models/ensemble/weighted/__init__.py,sha256=_LipTsDnYvTFmjZWsb1Vrm-eALsVVfUlF2gOpcaqE2Q,206
+autogluon/timeseries/models/ensemble/weighted/abstract.py,sha256=7vQVBK4TMBpESJ2EwnVklcljxmA2qWPQ9xpSREbtUwg,1543
+autogluon/timeseries/models/ensemble/weighted/basic.py,sha256=Kr8y0dlHRZg_q9AqBc3HIp1a5k_sXjrnQPlVi-63DCE,3066
+autogluon/timeseries/models/ensemble/weighted/greedy.py,sha256=zXJFenn1XxNNvCp4TlmIq1Dx3pUDWjKG1K3HsejmDeY,7323
+autogluon/timeseries/models/gluonts/__init__.py,sha256=YfyNYOkhhNsloA4MAavfmqKO29_q6o4lwPoV7L4_h7M,355
+autogluon/timeseries/models/gluonts/abstract.py,sha256=WKuUBy3ZF9VU87gaD9Us3c_xK2G1-XLeh1etVipf8hg,27769
+autogluon/timeseries/models/gluonts/dataset.py,sha256=wfEp5SPuB8bam7iTpX3Tf0FGdXp5vnZtpgC9G4VJ4tw,5111
+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=ASIZWBYs_cP0BwdrzHwblaNianPYcK5OqpqpiNxbxA0,11481
+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=PBnNhDXPJJatRIm9FXg9DXU_0ZkGSs2yvEqfaTwBVxM,12356
+autogluon/timeseries/models/toto/__init__.py,sha256=rQaVjZJV5ZsJGC0jhQ6CA4nYeXdV1KtlyDz2i2usQnY,54
+autogluon/timeseries/models/toto/dataloader.py,sha256=A5WHhnAe0J7fPo2KKG43hYLSrtUBGNweuqxMmClu3_A,3598
+autogluon/timeseries/models/toto/hf_pretrained_model.py,sha256=Q8bVUaSlQVE4xFn_v7H0h_NFTxzHiM1V17KFytc50jk,4783
+autogluon/timeseries/models/toto/model.py,sha256=3-5nR9qNqBFQLP6rNqBNlF4PBfnJHTcyjvz2GwdWwTg,8948
+autogluon/timeseries/models/toto/_internal/__init__.py,sha256=tKkiux9bD2Xu0AuVyTEx_sNOZutcluC7-d7tn7wsmec,193
+autogluon/timeseries/models/toto/_internal/dataset.py,sha256=xuAEOhoQNJGMoCxkLVLrgpdoOJuukAYbrSrnrkwFob0,6103
+autogluon/timeseries/models/toto/_internal/forecaster.py,sha256=UXiohiySn_Gs8kLheeVcVCO8qoEtYlEfMH1tukAOHsk,18520
+autogluon/timeseries/models/toto/_internal/backbone/__init__.py,sha256=hq5W62boH6HiEP8z3sHkI6_KM-Dd6TkDfWDm6DYE3J8,63
+autogluon/timeseries/models/toto/_internal/backbone/attention.py,sha256=HLUFoyqR8EqxUMT1BK-AjI4ClS8au35LcUo7Jx7Xhm0,9394
+autogluon/timeseries/models/toto/_internal/backbone/backbone.py,sha256=HUjpY2ZWed74UYKjp31erXF2ZHf3mmQMw_5_cCFeJGg,10104
+autogluon/timeseries/models/toto/_internal/backbone/distribution.py,sha256=8NXiaEVLuvjTW7L1t1RzooZFNERWv50zyLddbAwuYpo,2502
+autogluon/timeseries/models/toto/_internal/backbone/kvcache.py,sha256=QSVCrnbS2oD7wkJodZbP9XMVmrfCH6M3Zp44siF28Fg,5399
+autogluon/timeseries/models/toto/_internal/backbone/rope.py,sha256=Ghngo08DjHbwbyp6b-GXCyLeYR10dH-Y_RMOTYwIxPY,3527
+autogluon/timeseries/models/toto/_internal/backbone/scaler.py,sha256=opqyhHIZ6mPdPlrr3gA0qt9FFogIAYNDSq-P7CyQiqE,13728
+autogluon/timeseries/models/toto/_internal/backbone/transformer.py,sha256=5c-ngj4XHKlaedz1NkgdfQgqD2kUGkMn4mtGH_lTXsE,12410
+autogluon/timeseries/trainer/__init__.py,sha256=_tw3iioJfvtIV7wnjtEMv0yS8oabmCFxDnGRodYE7RI,72
+autogluon/timeseries/trainer/ensemble_composer.py,sha256=Vc8LfhGVUED70Y4DcIs3Jhpiur2EFXqVubgInixcb2I,9751
+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=yAHbpTjGKzVBepzepKuXEF5SvCQXDbsnyURV6mKLqaU,52002
+autogluon/timeseries/trainer/utils.py,sha256=_hSAWOYRZsp1qX2J6pJSxLrAAWwhVROc4_cvtfiTRzU,625
+autogluon/timeseries/transforms/__init__.py,sha256=fKlT4pkJ_8Gl7IUTc3uSDzt2Xow5iH5w6fPB3ePNrTg,127
+autogluon/timeseries/transforms/covariate_scaler.py,sha256=8E5DDRLUQ3SCNDR2Yw8FZDx7DnWVdokKhNNxbp_S-9I,7017
+autogluon/timeseries/transforms/target_scaler.py,sha256=tucfrWuXwTGv0WcJMo0bSk6--CkqGMDxiFPiUFl0RB8,6142
+autogluon/timeseries/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autogluon/timeseries/utils/features.py,sha256=GpemZRV7QiFRjZwP6NqpCVBg6m3KGBgp-eWUFzcpx54,22714
+autogluon/timeseries/utils/forecast.py,sha256=y3VV1rVCxOuh_p-2U9ftT_I5oU4gQQovxlw14jRGwyM,2259
+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.1b20251116.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon_timeseries-1.4.1b20251116.dist-info/licenses/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon_timeseries-1.4.1b20251116.dist-info/METADATA,sha256=2t9sW-KP5XI6x7WUxIkqXQqBCKRruP017VYThdeOPoA,12980
+autogluon_timeseries-1.4.1b20251116.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+autogluon_timeseries-1.4.1b20251116.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon_timeseries-1.4.1b20251116.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon_timeseries-1.4.1b20251116.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon_timeseries-1.4.1b20251116.dist-info/RECORD,,

{autogluon.timeseries-1.3.2b20250712.dist-info → autogluon_timeseries-1.4.1b20251116.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.45.1)
+Generator: setuptools (79.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any