autogluon.timeseries 1.1.2b20241119__py3-none-any.whl → 1.1.2b20241120__py3-none-any.whl

autogluon/timeseries/configs/presets_configs.py

@@ -10,6 +10,24 @@ TIMESERIES_PRESETS_CONFIGS = dict(
     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,

autogluon/timeseries/learner.py

@@ -32,6 +32,7 @@ class TimeSeriesLearner(AbstractLearner):
         eval_metric_seasonal_period: Optional[int] = None,
         prediction_length: int = 1,
         cache_predictions: bool = True,
+        ensemble_model_type: Optional[Type] = None,
         **kwargs,
     ):
         super().__init__(path_context=path_context)
@@ -44,6 +45,7 @@ class TimeSeriesLearner(AbstractLearner):
         self.quantile_levels = kwargs.get("quantile_levels", [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9])
         self.cache_predictions = cache_predictions
         self.freq: Optional[str] = None
+        self.ensemble_model_type = ensemble_model_type
 
         self.feature_generator = TimeSeriesFeatureGenerator(
             target=self.target, known_covariates_names=self.known_covariates_names
@@ -106,6 +108,7 @@ class TimeSeriesLearner(AbstractLearner):
                 val_splitter=val_splitter,
                 refit_every_n_windows=refit_every_n_windows,
                 cache_predictions=self.cache_predictions,
+                ensemble_model_type=self.ensemble_model_type,
             )
         )
         self.trainer = self.trainer_type(**trainer_init_kwargs)

autogluon/timeseries/models/abstract/abstract_timeseries_model.py

@@ -67,6 +67,7 @@ class AbstractTimeSeriesModel(AbstractModel):
     _oof_filename = "oof.pkl"
     # TODO: For which models should we override this parameter?
     _covariate_regressor_fit_time_fraction: float = 0.5
+    default_max_time_limit_ratio: float = 0.9
 
     # TODO: refactor "pruned" methods after AbstractModel is refactored
     predict_proba = None
@@ -174,6 +175,11 @@ class AbstractTimeSeriesModel(AbstractModel):
             self._oof_predictions = self.load_oof_predictions(self.path)
         return self._oof_predictions
 
+    def _get_default_auxiliary_params(self) -> dict:
+        default_auxiliary_params = super()._get_default_auxiliary_params()
+        default_auxiliary_params["max_time_limit_ratio"] = self.default_max_time_limit_ratio
+        return default_auxiliary_params
+
     def _initialize(self, **kwargs) -> None:
         self._init_params_aux()
         self._init_params()

autogluon/timeseries/models/chronos/model.py

@@ -1,7 +1,6 @@
 import logging
 import os
 import shutil
-import time
 from pathlib import Path
 from typing import Any, Dict, Literal, Optional, Union
 
@@ -68,6 +67,7 @@ MODEL_ALIASES = {
     "small": "autogluon/chronos-t5-small",
     "base": "autogluon/chronos-t5-base",
     "large": "autogluon/chronos-t5-large",
+    "bolt-tiny": "autogluon/chronos-bolt-tiny",
     "bolt-mini": "autogluon/chronos-bolt-mini",
     "bolt-small": "autogluon/chronos-bolt-small",
     "bolt-base": "autogluon/chronos-bolt-base",
@@ -113,10 +113,12 @@ class ChronosModel(AbstractTimeSeriesModel):
     batch_size : int, default = 16
         Size of batches used during inference
     num_samples : int, default = 20
-        Number of samples used during inference
+        Number of samples used during inference, only used for the original Chronos models
     device : str, default = None
         Device to use for inference (and fine-tuning, if enabled). If None, model will use the GPU if available.
-        For larger model sizes `small`, `base`, and `large`; inference will fail if no GPU is available.
+        For larger Chronos model sizes ``small``, ``base``, and ``large``; inference will fail if no GPU is available.
+        For Chronos-Bolt models, inference can be done on the CPU. Although fine-tuning the smaller Chronos models
+        (``tiny`` and ``mini``) and all Chronos-Bolt is allowed on the CPU, we recommend using a GPU for faster fine-tuning.
     context_length : int or None, default = None
         The context length to use in the model. Shorter context lengths will decrease model accuracy, but result
         in faster inference. If None, the model will infer context length from the data set length at inference
@@ -140,9 +142,9 @@ class ChronosModel(AbstractTimeSeriesModel):
         If True, the pretrained model will be fine-tuned
     fine_tune_lr: float, default = 0.0001
         The learning rate used for fine-tuning
-    fine_tune_steps : int, default = 5000
+    fine_tune_steps : int, default = 1000
         The number of gradient update steps to fine-tune for
-    fine_tune_batch_size : int, default = 16
+    fine_tune_batch_size : int, default = 32
         The batch size to use for fine-tuning
     fine_tune_shuffle_buffer_size : int, default = 10000
         The size of the shuffle buffer to shuffle the data during fine-tuning. If None, shuffling will
@@ -162,6 +164,7 @@ class ChronosModel(AbstractTimeSeriesModel):
     # default number of samples for prediction
     default_num_samples: int = 20
     default_model_path = "autogluon/chronos-t5-small"
+    default_max_time_limit_ratio = 0.8
     maximum_context_length = 2048
     fine_tuned_ckpt_name: str = "fine-tuned-ckpt"
 
@@ -216,7 +219,6 @@ class ChronosModel(AbstractTimeSeriesModel):
         )
 
         self.model_pipeline: Optional[Any] = None  # of type BaseChronosPipeline
-        self.time_limit: Optional[float] = None
 
     def save(self, path: str = None, verbose: bool = True) -> str:
         pipeline = self.model_pipeline
@@ -234,7 +236,7 @@ class ChronosModel(AbstractTimeSeriesModel):
 
         fine_tune_ckpt_path = Path(model.path) / cls.fine_tuned_ckpt_name
         if fine_tune_ckpt_path.exists():
-            logger.debug(f"Fine-tuned checkpoint exists, setting model_path to {fine_tune_ckpt_path}")
+            logger.debug(f"\tFine-tuned checkpoint exists, setting model_path to {fine_tune_ckpt_path}")
             model.model_path = fine_tune_ckpt_path
 
         return model
@@ -320,8 +322,8 @@ class ChronosModel(AbstractTimeSeriesModel):
         init_args.setdefault("fine_tune", False)
         init_args.setdefault("keep_transformers_logs", False)
         init_args.setdefault("fine_tune_lr", 1e-4)
-        init_args.setdefault("fine_tune_steps", 5000)
-        init_args.setdefault("fine_tune_batch_size", self.default_batch_size)
+        init_args.setdefault("fine_tune_steps", 1000)
+        init_args.setdefault("fine_tune_batch_size", 32)
         init_args.setdefault("eval_during_fine_tune", False)
         init_args.setdefault("fine_tune_eval_max_items", 256)
         init_args.setdefault("fine_tune_shuffle_buffer_size", 10_000)
@@ -399,7 +401,6 @@ class ChronosModel(AbstractTimeSeriesModel):
 
         eval_during_fine_tune = val_data is not None and fine_tune_args["eval_during_fine_tune"]
 
-        start_time = time.monotonic()
         if do_fine_tune:
             context_length = self._get_context_length(train_data)
             # load model pipeline to device memory
@@ -428,7 +429,7 @@ class ChronosModel(AbstractTimeSeriesModel):
 
                 if self.prediction_length != fine_tune_prediction_length:
                     logger.debug(
-                        f"ChronosBolt models can only be fine-tuned with a maximum prediction_length of {model_prediction_length}. "
+                        f"\tChronosBolt models can only be fine-tuned with a maximum prediction_length of {model_prediction_length}. "
                         f"Fine-tuning prediction_length has been changed to {fine_tune_prediction_length}."
                     )
 
@@ -436,10 +437,15 @@ class ChronosModel(AbstractTimeSeriesModel):
             fine_tune_trainer_kwargs["disable_tqdm"] = fine_tune_trainer_kwargs.get("disable_tqdm", (verbosity < 3))
             fine_tune_trainer_kwargs["use_cpu"] = str(self.model_pipeline.inner_model.device) == "cpu"
 
-            # TODO: adamw_torch_fused is not supported on CPU in torch <= 2.3. When torch 2.4 becomes the lower bound
-            # this if block can be removed because torch >= 2.4 supports AdamW optimizer with fused=True on CPU
-            if fine_tune_trainer_kwargs["use_cpu"] and fine_tune_trainer_kwargs["optim"] == "adamw_torch_fused":
-                fine_tune_trainer_kwargs["optim"] = "adamw_torch"
+            if fine_tune_trainer_kwargs["use_cpu"]:
+                logger.info(
+                    "\tFine-tuning on the CPU detected. We recommend using a GPU for faster fine-tuning of Chronos."
+                )
+
+                # TODO: adamw_torch_fused is not supported on CPU in torch <= 2.3. When torch 2.4 becomes the lower bound
+                # this if block can be removed because torch >= 2.4 supports AdamW optimizer with fused=True on CPU
+                if fine_tune_trainer_kwargs["optim"] == "adamw_torch_fused":
+                    fine_tune_trainer_kwargs["optim"] = "adamw_torch"
 
             output_dir = Path(fine_tune_trainer_kwargs["output_dir"])
 
@@ -509,37 +515,16 @@ class ChronosModel(AbstractTimeSeriesModel):
                 )
                 trainer.add_callback(LoggerCallback())
 
-            if val_data is not None:
-                # evaluate once before training
-                zero_shot_eval_loss = trainer.evaluate()["eval_loss"]
-
             trainer.train()
 
-            if eval_during_fine_tune:
-                # get the best eval_loss logged during fine-tuning
-                log_history_df = pd.DataFrame(trainer.state.log_history)
-                best_train_eval_loss = log_history_df["eval_loss"].min()
-            elif val_data is not None:
-                # evaluate at the end of fine-tuning
-                best_train_eval_loss = trainer.evaluate()["eval_loss"]
-
-            if val_data is None or best_train_eval_loss <= zero_shot_eval_loss:
-                fine_tuned_ckpt_path = Path(self.path) / self.fine_tuned_ckpt_name
-                logger.info(f"Saving fine-tuned model to {fine_tuned_ckpt_path}")
-                self.model_pipeline.inner_model.save_pretrained(Path(self.path) / self.fine_tuned_ckpt_name)
-            else:
-                # Reset the model to its pretrained state
-                logger.info("Validation loss worsened after fine-tuning. Reverting to the pretrained model.")
-                self.model_pipeline = None
-                self.load_model_pipeline(is_training=False)
+            fine_tuned_ckpt_path = Path(self.path) / self.fine_tuned_ckpt_name
+            logger.info(f"\tSaving fine-tuned model to {fine_tuned_ckpt_path}")
+            self.model_pipeline.inner_model.save_pretrained(Path(self.path) / self.fine_tuned_ckpt_name)
 
             if not fine_tune_args["keep_transformers_logs"]:
                 logger.debug(f"Removing transformers_logs directory {output_dir}")
                 shutil.rmtree(output_dir)
 
-        if time_limit is not None:
-            self.time_limit = time_limit - (time.monotonic() - start_time)  # inference time budget
-
     def _get_inference_data_loader(
         self,
         data: TimeSeriesDataFrame,
@@ -635,16 +620,3 @@ class ChronosModel(AbstractTimeSeriesModel):
             "can_use_train_data": do_fine_tune,
             "can_use_val_data": do_fine_tune,
         }
-
-    def score_and_cache_oof(
-        self,
-        val_data: TimeSeriesDataFrame,
-        store_val_score: bool = False,
-        store_predict_time: bool = False,
-        **predict_kwargs,
-    ) -> None:
-        # All computation happens during inference, so we provide the time_limit at prediction time
-        # TODO: Once custom predict_kwargs is allowed, make sure that `time_limit` is not among the keys
-        super().score_and_cache_oof(
-            val_data, store_val_score, store_predict_time, time_limit=self.time_limit, **predict_kwargs
-        )

autogluon/timeseries/models/chronos/pipeline/base.py

@@ -135,6 +135,7 @@ class BaseChronosPipeline(metaclass=PipelineRegistry):
         """
         from transformers import AutoConfig
 
+        kwargs.setdefault("resume_download", None)  # silence huggingface_hub warning
         if str(pretrained_model_name_or_path).startswith("s3://"):
             from .utils import cache_model_from_s3
 

autogluon/timeseries/models/chronos/pipeline/chronos.py

@@ -546,6 +546,7 @@ class ChronosPipeline(BaseChronosPipeline):
                 "onnx",
                 "openvino",
             ], "optimization_strategy not recognized. Please provide one of `onnx` or `openvino`"
+            kwargs.pop("resume_download", None)  # Optimized pipeline does not support 'resume_download' kwargs
             torch_dtype = kwargs.pop("torch_dtype", "auto")
             if torch_dtype != "auto":
                 logger.warning(f"\t`torch_dtype` will be ignored for optimization_strategy {optimization_strategy}")

autogluon/timeseries/models/chronos/pipeline/utils.py

@@ -317,7 +317,8 @@ class TimeLimitCallback(TrainerCallback):
     def on_step_end(self, args, state, control, **kwargs):
         elapsed_time = time.monotonic() - self.start_time
         if elapsed_time > self.time_limit:
-            raise TimeLimitExceeded
+            logger.info("\tStopping fine-tuning since time_limit is reached")
+            control.should_training_stop = True
 
 
 class LoggerCallback(TrainerCallback):

autogluon/timeseries/models/local/abstract_local_model.py

@@ -44,6 +44,7 @@ class AbstractLocalModel(AbstractTimeSeriesModel):
     allowed_local_model_args: List[str] = []
     default_n_jobs: Union[int, float] = AG_DEFAULT_N_JOBS
     default_max_ts_length: Optional[int] = 2500
+    default_max_time_limit_ratio = 1.0
     init_time_in_seconds: int = 0
 
     def __init__(
@@ -84,7 +85,6 @@ class AbstractLocalModel(AbstractTimeSeriesModel):
 
         self._local_model_args: Dict[str, Any] = None
         self._seasonal_period: Optional[int] = None
-        self.time_limit: Optional[float] = None
         self._dummy_forecast: Optional[pd.DataFrame] = None
 
     @property
@@ -138,7 +138,6 @@ class AbstractLocalModel(AbstractTimeSeriesModel):
         self._seasonal_period = local_model_args["seasonal_period"]
 
         self._local_model_args = self._update_local_model_args(local_model_args=local_model_args)
-        self.time_limit = time_limit
 
         self._dummy_forecast = self._get_dummy_forecast(train_data)
         return self
@@ -187,18 +186,6 @@ class AbstractLocalModel(AbstractTimeSeriesModel):
         predictions_df.index = get_forecast_horizon_index_ts_dataframe(data, self.prediction_length, freq=self.freq)
         return TimeSeriesDataFrame(predictions_df)
 
-    def score_and_cache_oof(
-        self,
-        val_data: TimeSeriesDataFrame,
-        store_val_score: bool = False,
-        store_predict_time: bool = False,
-        **predict_kwargs,
-    ) -> None:
-        # All computation happens during inference, so we provide the time_limit at prediction time
-        super().score_and_cache_oof(
-            val_data, store_val_score, store_predict_time, time_limit=self.time_limit, **predict_kwargs
-        )
-
     def _predict_wrapper(self, time_series: pd.Series, end_time: Optional[float] = None) -> Tuple[pd.DataFrame, bool]:
         if end_time is not None and time.time() >= end_time:
             raise TimeLimitExceeded

autogluon/timeseries/models/multi_window/multi_window_model.py

@@ -33,6 +33,7 @@ class MultiWindowBacktestingModel(AbstractTimeSeriesModel):
     """
 
     # TODO: Remove the MultiWindowBacktestingModel class, move the logic to AbstractTimeSeriesTrainer
+    default_max_time_limit_ratio = 1.0
 
     def __init__(
         self,
@@ -124,8 +125,7 @@ class MultiWindowBacktestingModel(AbstractTimeSeriesModel):
                     num_refits_remaining = math.ceil(
                         (val_splitter.num_val_windows - window_index) / refit_every_n_windows
                     )
-                    # Reserve 10% of the remaining time for prediction, use 90% of time for training
-                    time_left_for_window = 0.9 * time_left / num_refits_remaining
+                    time_left_for_window = time_left / num_refits_remaining
 
             if refit_this_window:
                 model = self.get_child_model(window_index)
@@ -138,7 +138,15 @@ class MultiWindowBacktestingModel(AbstractTimeSeriesModel):
                 )
                 model.fit_time = time.time() - model_fit_start_time
                 most_recent_refit_window = f"W{window_index}"
-            model.score_and_cache_oof(val_fold, store_val_score=True, store_predict_time=True)
+
+            if time_limit is None:
+                time_left_for_prediction = None
+            else:
+                time_left_for_prediction = time_limit - (time.time() - global_fit_start_time)
+
+            model.score_and_cache_oof(
+                val_fold, store_val_score=True, store_predict_time=True, time_limit=time_left_for_prediction
+            )
 
             oof_predictions_per_window.append(model.get_oof_predictions()[0])
 

autogluon/timeseries/models/presets.py

@@ -7,6 +7,7 @@ from typing import Any, Dict, List, Optional, Type, Union
 from autogluon.common import space
 from autogluon.core import constants
 from autogluon.timeseries.metrics import TimeSeriesScorer
+from autogluon.timeseries.utils.features import CovariateMetadata
 
 from . import (
     ADIDAModel,
@@ -133,6 +134,7 @@ def get_default_hps(key):
             "RecursiveTabular": {},
             "DirectTabular": {},
             "TemporalFusionTransformer": {},
+            "Chronos": {"model_path": "bolt-small"},
         },
         "light_inference": {
             "SeasonalNaive": {},
@@ -145,18 +147,33 @@ def get_default_hps(key):
             "SeasonalNaive": {},
             "Croston": {},
             "AutoETS": {},
-            "AutoARIMA": {},
             "NPTS": {},
             "DynamicOptimizedTheta": {},
-            # TODO: Define separate model for each tabular submodel?
-            "RecursiveTabular": {
-                "tabular_hyperparameters": {"NN_TORCH": {"proc.impute_strategy": "constant"}, "GBM": {}},
-            },
+            "RecursiveTabular": {},
             "DirectTabular": {},
             "TemporalFusionTransformer": {},
             "PatchTST": {},
             "DeepAR": {},
-            "Chronos": {"model_path": "base"},
+            "Chronos": [
+                {
+                    "ag_args": {"name_suffix": "ZeroShot"},
+                    "model_path": "bolt-base",
+                },
+                {
+                    "ag_args": {"name_suffix": "FineTuned"},
+                    "model_path": "bolt-small",
+                    "fine_tune": True,
+                    "target_scaler": "standard",
+                    "covariate_regressor": {"model_name": "CAT", "model_hyperparameters": {"iterations": 1_000}},
+                },
+            ],
+            "TiDE": {
+                "encoder_hidden_dim": 256,
+                "decoder_hidden_dim": 256,
+                "temporal_hidden_dim": 64,
+                "num_batches_per_epoch": 100,
+                "lr": 1e-4,
+            },
         },
     }
     return default_model_hps[key]
@@ -170,6 +187,7 @@ def get_preset_models(
     eval_metric_seasonal_period: Optional[int],
     hyperparameters: Union[str, Dict, None],
     hyperparameter_tune: bool,
+    metadata: CovariateMetadata,
     all_assigned_names: List[str],
     excluded_model_types: List[str],
     multi_window: bool = False,
@@ -247,6 +265,7 @@ def get_preset_models(
                 prediction_length=prediction_length,
                 eval_metric=eval_metric,
                 eval_metric_seasonal_period=eval_metric_seasonal_period,
+                metadata=metadata,
                 hyperparameters=model_hps,
                 **kwargs,
             )

autogluon/timeseries/predictor.py

@@ -217,6 +217,7 @@ class TimeSeriesPredictor(TimeSeriesPredictorDeprecatedMixin):
                 prediction_length=self.prediction_length,
                 quantile_levels=self.quantile_levels,
                 cache_predictions=self.cache_predictions,
+                ensemble_model_type=kwargs.pop("ensemble_model_type", None),
             )
         )
         # Using `TimeSeriesLearner` as default argument breaks doc generation with Sphnix

autogluon/timeseries/trainer/abstract_trainer.py

@@ -29,7 +29,7 @@ from autogluon.timeseries.utils.features import (
     CovariateMetadata,
     PermutationFeatureImportanceTransform,
 )
-from autogluon.timeseries.utils.warning_filters import disable_tqdm
+from autogluon.timeseries.utils.warning_filters import disable_tqdm, warning_filter
 
 logger = logging.getLogger("autogluon.timeseries.trainer")
 
@@ -264,6 +264,7 @@ class AbstractTimeSeriesTrainer(SimpleAbstractTrainer):
         val_splitter: Optional[AbstractWindowSplitter] = None,
         refit_every_n_windows: Optional[int] = 1,
         cache_predictions: bool = True,
+        ensemble_model_type: Optional[Type] = None,
         **kwargs,
     ):
         super().__init__(path=path, save_data=save_data, low_memory=True, **kwargs)
@@ -276,7 +277,13 @@ class AbstractTimeSeriesTrainer(SimpleAbstractTrainer):
         self.skip_model_selection = skip_model_selection
         # Ensemble cannot be fit if val_scores are not computed
         self.enable_ensemble = enable_ensemble and not skip_model_selection
-        self.ensemble_model_type = TimeSeriesGreedyEnsemble
+        if ensemble_model_type is None:
+            ensemble_model_type = TimeSeriesGreedyEnsemble
+        else:
+            logger.warning(
+                "Using a custom `ensemble_model_type` is experimental functionality that may break in future versions."
+            )
+        self.ensemble_model_type = ensemble_model_type
 
         self.verbosity = verbosity
 
@@ -519,8 +526,12 @@ class AbstractTimeSeriesTrainer(SimpleAbstractTrainer):
             fit_end_time = time.time()
             model.fit_time = model.fit_time or (fit_end_time - fit_start_time)
 
+            if time_limit is not None:
+                time_limit = fit_end_time - fit_start_time
             if val_data is not None and not self.skip_model_selection:
-                model.score_and_cache_oof(val_data, store_val_score=True, store_predict_time=True)
+                model.score_and_cache_oof(
+                    val_data, store_val_score=True, store_predict_time=True, time_limit=time_limit
+                )
 
             self._log_scores_and_times(model.val_score, model.fit_time, model.predict_time)
 
@@ -736,7 +747,8 @@ class AbstractTimeSeriesTrainer(SimpleAbstractTrainer):
             quantile_levels=self.quantile_levels,
             metadata=self.metadata,
         )
-        ensemble.fit_ensemble(model_preds, data_per_window=data_per_window, time_limit=time_limit)
+        with warning_filter():
+            ensemble.fit_ensemble(model_preds, data_per_window=data_per_window, time_limit=time_limit)
         ensemble.fit_time = time.time() - time_start
 
         predict_time = 0

autogluon/timeseries/version.py

@@ -1,3 +1,3 @@
 """This is the autogluon version file."""
-__version__ = '1.1.2b20241119'
+__version__ = '1.1.2b20241120'
 __lite__ = False

{autogluon.timeseries-1.1.2b20241119.dist-info → autogluon.timeseries-1.1.2b20241120.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 1.1.2b20241119
+Version: 1.1.2b20241120
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -53,9 +53,9 @@ 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.1.2b20241119
-Requires-Dist: autogluon.common==1.1.2b20241119
-Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20241119
+Requires-Dist: autogluon.core[raytune]==1.1.2b20241120
+Requires-Dist: autogluon.common==1.1.2b20241120
+Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20241120
 Provides-Extra: all
 Requires-Dist: optimum[onnxruntime]<1.20,>=1.17; extra == "all"
 Provides-Extra: chronos-onnx

{autogluon.timeseries-1.1.2b20241119.dist-info → autogluon.timeseries-1.1.2b20241120.dist-info}/RECORD

@@ -1,13 +1,13 @@
-autogluon.timeseries-1.1.2b20241119-py3.8-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.timeseries-1.1.2b20241120-py3.8-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=3dUxI-U6TGfNtRQUzWTvBIo1GKeXYOhxIX_q7Fed9eA,14013
-autogluon/timeseries/predictor.py,sha256=R9m-TYmlA4WoJbdYEL_AnEM26EhRIclynOfSmpO7mBk,84926
+autogluon/timeseries/learner.py,sha256=mFnBC750C5PqgkkYNYni9oYQ5a6K8pXSsDLRDXuA7DI,14182
+autogluon/timeseries/predictor.py,sha256=EsJAkzlEkCFxYO55BTfjlaJPNwQwnS6yyj5nXjMr3XQ,85003
 autogluon/timeseries/regressor.py,sha256=tqQ2zWImxpuEyaAM0DeCjOZ-xcWUYZbCXsqd471xXxQ,8351
 autogluon/timeseries/splitter.py,sha256=eghGwAAN2_cxGk5aJBILgjGWtLzjxJcytMy49gg_q18,3061
-autogluon/timeseries/version.py,sha256=0242cKF96lnZcXGbtRmKZUi3yVY1-rAip2PtRhF89WQ,90
+autogluon/timeseries/version.py,sha256=t6TdlyVVjsI-etjHY5tqRoGECcEKxWQwAjz-UboJ0sE,90
 autogluon/timeseries/configs/__init__.py,sha256=BTtHIPCYeGjqgOcvqb8qPD4VNX-ICKOg6wnkew1cPOE,98
-autogluon/timeseries/configs/presets_configs.py,sha256=94-yL9teDHKs2irWjP3kpewI7FE1ChYCgEgz9XHJ6gc,1965
+autogluon/timeseries/configs/presets_configs.py,sha256=k5RRP0DQMa2Xq2oWAbRKouWUH8xUuBx2tIWk-we6_I8,2543
 autogluon/timeseries/dataset/__init__.py,sha256=UvnhAN5tjgxXTHoZMQDy64YMDj4Xxa68yY7NP4vAw0o,81
 autogluon/timeseries/dataset/ts_dataframe.py,sha256=9bJQeg3HkPeVnyxzwqAJiTJGYXths7vxUV_3-OsJ6pk,48640
 autogluon/timeseries/metrics/__init__.py,sha256=LLGmYaexsx7CregV-QaHc5exjZbsJfBSVOtxHRGC0ho,2139
@@ -16,21 +16,21 @@ autogluon/timeseries/metrics/point.py,sha256=b19Ed4dS_ROdkrOZIik_Q3-8deCN9IQSZXt
 autogluon/timeseries/metrics/quantile.py,sha256=eemdLbo3y2wstnVkuA-f55YXywctUmSW1EhIW4BsoH4,3965
 autogluon/timeseries/metrics/utils.py,sha256=HuDe1BNe8yJU4f_DKM913nNrUueoRaw6zhxm1-S20s0,910
 autogluon/timeseries/models/__init__.py,sha256=MYD9JJ-wUDE5B6jW6E6LU2eXQ6vflfQBvqQJkdzJa3A,1189
-autogluon/timeseries/models/presets.py,sha256=ujNt_hft_5eNkh-Wj_Na9GBdBmI-JdnBnOEHq8X0qXc,11778
+autogluon/timeseries/models/presets.py,sha256=toZePXy1UYKIoQFQnXUIdhZuq7DBNMyfsYU_FwV94Nk,12473
 autogluon/timeseries/models/abstract/__init__.py,sha256=wvDsQAZIV0N3AwBeMaGItoQ82trEfnT-nol2AAOIxBg,102
-autogluon/timeseries/models/abstract/abstract_timeseries_model.py,sha256=B1R0PBymUAwhIGkPIt29X-J9o9Ipdu-bQR0gK-nmcRU,30320
+autogluon/timeseries/models/abstract/abstract_timeseries_model.py,sha256=82lg2odAxzyhM3dkV3Msqv8AeYLILnnk-UvNnYobzFA,30628
 autogluon/timeseries/models/abstract/model_trial.py,sha256=ENPg_7nsdxIvaNM0o0UShZ3x8jFlRmwRc5m0fGPC0TM,3720
 autogluon/timeseries/models/autogluon_tabular/__init__.py,sha256=r9i6jWcyeLHYClkcMSKRVsfrkBUMxpDrTATNTBc_qgQ,136
 autogluon/timeseries/models/autogluon_tabular/mlforecast.py,sha256=vfWXLdxYlbzjKJa1qrFN-qzxgG2tiWxOSBOnmnNVxtA,33295
 autogluon/timeseries/models/autogluon_tabular/transforms.py,sha256=XVoy8KpvoeX38lHHAXq4Be9LCxKjxZ36SOFeSAICRFM,2524
 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=pkT-V6yYCxz6TTgn1mQ5QHNTdpyn_wyj7jM80vnrDFQ,30270
+autogluon/timeseries/models/chronos/model.py,sha256=krsF5Fu-Q_FYTW-WbeXX4--s1ZH5kY-LlxvZg2jvfT0,29085
 autogluon/timeseries/models/chronos/pipeline/__init__.py,sha256=N-YZH9BGBoi99r5cznJe1zEEjwjIg7cOYIHZkKuJq44,247
-autogluon/timeseries/models/chronos/pipeline/base.py,sha256=aAXCKy7Jmip4BI2UdPMoPe2gdDMbJHKxEolcTx_5SYQ,5463
-autogluon/timeseries/models/chronos/pipeline/chronos.py,sha256=one9UwXja042JeXYSdenpCoIQ2DWZKpKiivnom5Epbw,22046
+autogluon/timeseries/models/chronos/pipeline/base.py,sha256=HlWQTS5q7UMzwbA5Pmg_N01AxuGfTf2tP5xq2jgavqI,5549
+autogluon/timeseries/models/chronos/pipeline/chronos.py,sha256=doAaWbrfNilkP9ORtjDnL-1S5ge4sOKhzGN-mgsY2bM,22158
 autogluon/timeseries/models/chronos/pipeline/chronos_bolt.py,sha256=2MJuik-YFgONZ3X2DciAph5So6ABys5ppQhBC81gLyk,20083
-autogluon/timeseries/models/chronos/pipeline/utils.py,sha256=_P_9m9Wl4FC2QyoKLluT4l7FLmZU2xw6G4xNcUpZE4k,13043
+autogluon/timeseries/models/chronos/pipeline/utils.py,sha256=hw4QbRy81V9Jkndda9p_KgO9CrtYKbNq5kp0r8m0Rek,13133
 autogluon/timeseries/models/ensemble/__init__.py,sha256=kFr11Gmt7lQJu9Rr8HuIPphQN5l1TsoorfbJm_O3a_s,128
 autogluon/timeseries/models/ensemble/abstract_timeseries_ensemble.py,sha256=tifETwmiEGt-YtQ9eNK7ojJ3fBvtFMUJvisbfkIJ7gw,3393
 autogluon/timeseries/models/ensemble/greedy_ensemble.py,sha256=5HvZuW5osgsZg3V69k82nKEOy_YgeH1JTfQa7F3cU7s,7220
@@ -39,14 +39,14 @@ autogluon/timeseries/models/gluonts/abstract_gluonts.py,sha256=M4vGs5oNZqr_ebE4U
 autogluon/timeseries/models/gluonts/torch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/timeseries/models/gluonts/torch/models.py,sha256=Pu7f43jr1C5S3k_bVqRB8ENuBHNEWT4ssUTdZoA1J58,25556
 autogluon/timeseries/models/local/__init__.py,sha256=e2UImoJhmj70E148IIObv90C_bHxgyLNk6YsS4p7pfs,701
-autogluon/timeseries/models/local/abstract_local_model.py,sha256=WBBJp2h2UMUVuguTrUmJOI21neil-ZnGGZDy-zJPD2M,12592
+autogluon/timeseries/models/local/abstract_local_model.py,sha256=5WI-TApbWi01bsQghnQaPH0SXp5SwV5VAPBwmbCTbeU,12090
 autogluon/timeseries/models/local/naive.py,sha256=iwRcFMFmJKPWPbD9TWaIUS51oav69F_VAp6-jb_5SUE,7249
 autogluon/timeseries/models/local/npts.py,sha256=Bp74doKnfpGE8ywP4FWOCI_RwRMsmgocYDfGtq764DA,4143
 autogluon/timeseries/models/local/statsforecast.py,sha256=cFJ_A7LR2jTmFNGgMxt3xvEivQVYuV6bDCMii8-TKH0,32424
 autogluon/timeseries/models/multi_window/__init__.py,sha256=Bq7AT2Jxdd4WNqmjTdzeqgNiwn1NCyWp4tBIWaM-zfI,60
-autogluon/timeseries/models/multi_window/multi_window_model.py,sha256=mTxqPBsN0Ri2c56MYZU6pMF9dDFyEl5eHmhqHPe3ouw,11596
+autogluon/timeseries/models/multi_window/multi_window_model.py,sha256=2o_wKNCNP570c0UuNufPhEDYOWQ2mFYwmmHUZzPneRI,11792
 autogluon/timeseries/trainer/__init__.py,sha256=lxiOT-Gc6BEnr_yWQqra85kEngeM_wtH2SCaRbmC_qE,170
-autogluon/timeseries/trainer/abstract_trainer.py,sha256=zCzzVO8yssqSyteMoUkDT5zWi44Oc91pRWiRdQR1We8,60521
+autogluon/timeseries/trainer/abstract_trainer.py,sha256=3DEr4hATu2zNzkuzwO5UeqS25Ak4k5w4OFAtCmYqmAc,61053
 autogluon/timeseries/trainer/auto_trainer.py,sha256=psJFZBwWWPlLjNwAgvO4OUJXsRW1sTN2YS9a4pdoeoE,3344
 autogluon/timeseries/transforms/__init__.py,sha256=Stym_998LZQgKPuFN4_w1AcJFh4_AeaQLXgXLzv53kY,299
 autogluon/timeseries/transforms/covariate_scaler.py,sha256=iscshgfNTCn379Q73BJXyDUFFm1WRclzKdG2MIPTOEc,6587
@@ -60,11 +60,11 @@ autogluon/timeseries/utils/datetime/base.py,sha256=3NdsH3NDq4cVAOSoy3XpaNixyNlbj
 autogluon/timeseries/utils/datetime/lags.py,sha256=GoLtvcZ8oKb3QkoBJ9E59LSPLOP7Qjxrr2UmMSZgjyw,5909
 autogluon/timeseries/utils/datetime/seasonality.py,sha256=h_4w00iEytAz_N_EpCENQ8RCXy7KQITczrYjBgVqWkQ,764
 autogluon/timeseries/utils/datetime/time_features.py,sha256=PAXbYbQ0z_5GFbkxSNi41zLY_2-U3x0Ynm1m_WhdtGc,2572
-autogluon.timeseries-1.1.2b20241119.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.timeseries-1.1.2b20241119.dist-info/METADATA,sha256=I9IrNzlXQskc4xcwgmngBRT6iIDZBdmd0rvXxOWaXcQ,12388
-autogluon.timeseries-1.1.2b20241119.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.timeseries-1.1.2b20241119.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
-autogluon.timeseries-1.1.2b20241119.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.timeseries-1.1.2b20241119.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.timeseries-1.1.2b20241119.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.timeseries-1.1.2b20241119.dist-info/RECORD,,
+autogluon.timeseries-1.1.2b20241120.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.timeseries-1.1.2b20241120.dist-info/METADATA,sha256=gf3xf53SAhlU4E2tfn_mQ9Fj5aclpX1ZXczq6_6ABRk,12388
+autogluon.timeseries-1.1.2b20241120.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.timeseries-1.1.2b20241120.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+autogluon.timeseries-1.1.2b20241120.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-1.1.2b20241120.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-1.1.2b20241120.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.timeseries-1.1.2b20241120.dist-info/RECORD,,