autogluon.timeseries 1.1.2b20241121__py3-none-any.whl → 1.1.2b20241123__py3-none-any.whl

autogluon/timeseries/models/chronos/model.py

@@ -1,6 +1,7 @@
 import logging
 import os
 import shutil
+import warnings
 from pathlib import Path
 from typing import Any, Dict, Literal, Optional, Union
 
@@ -15,6 +16,11 @@ from autogluon.timeseries.utils.warning_filters import disable_duplicate_logs, w
 
 logger = logging.getLogger("autogluon.timeseries.models.chronos")
 
+# TODO: Replace `evaluation_strategy` with `eval_strategy` when upgrading to `transformers>=4.41` + remove warning filter
+warnings.filterwarnings("ignore", category=FutureWarning, message="`evaluation_strategy` is deprecated")
+# TODO: Remove warning filter when upgrading to `transformers>=4.40`
+warnings.filterwarnings("ignore", category=FutureWarning, message="Passing the following arguments to ")
+
 
 # allowed HuggingFace model paths with custom parameter definitions
 MODEL_CONFIGS = {
@@ -104,7 +110,7 @@ class ChronosModel(AbstractTimeSeriesModel):
 
     Other Parameters
     ----------------
-    model_path: str, default = "autogluon/chronos-t5-small"
+    model_path: str, default = "autogluon/chronos-bolt-small"
         Model path used for the model, i.e., a HuggingFace transformers ``name_or_path``. Can be a
         compatible model name on HuggingFace Hub or a local path to a model directory. Original
         Chronos models (i.e., ``autogluon/chronos-t5-{model_size}``) can be specified with aliases
@@ -140,8 +146,9 @@ class ChronosModel(AbstractTimeSeriesModel):
         for more information.
     fine_tune : bool, default = False
         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_lr: float, default = 1e-5
+        The learning rate used for fine-tuning. This default is suitable for Chronos-Bolt models; for the original
+        Chronos models, we recommend using a higher learning rate such as ``1e-4``
     fine_tune_steps : int, default = 1000
         The number of gradient update steps to fine-tune for
     fine_tune_batch_size : int, default = 32
@@ -163,7 +170,7 @@ class ChronosModel(AbstractTimeSeriesModel):
 
     # default number of samples for prediction
     default_num_samples: int = 20
-    default_model_path = "autogluon/chronos-t5-small"
+    default_model_path = "autogluon/chronos-bolt-small"
     default_max_time_limit_ratio = 0.8
     maximum_context_length = 2048
     fine_tuned_ckpt_name: str = "fine-tuned-ckpt"
@@ -198,7 +205,7 @@ class ChronosModel(AbstractTimeSeriesModel):
         self.context_length = hyperparameters.get("context_length")
 
         if self.context_length is not None and self.context_length > self.maximum_context_length:
-            logger.warning(
+            logger.info(
                 f"\tContext length {self.context_length} exceeds maximum context length {self.maximum_context_length}."
                 f"Context length will be set to {self.maximum_context_length}."
             )
@@ -321,7 +328,7 @@ 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_lr", 1e-5)
         init_args.setdefault("fine_tune_steps", 1000)
         init_args.setdefault("fine_tune_batch_size", 32)
         init_args.setdefault("eval_during_fine_tune", False)
@@ -341,6 +348,7 @@ class ChronosModel(AbstractTimeSeriesModel):
             logging_dir=str(output_dir),
             logging_strategy="steps",
             logging_steps=100,
+            disable_tqdm=True,
             report_to="none",
             max_steps=init_args["fine_tune_steps"],
             gradient_accumulation_steps=1,
@@ -434,7 +442,6 @@ class ChronosModel(AbstractTimeSeriesModel):
                     )
 
             fine_tune_trainer_kwargs = fine_tune_args["fine_tune_trainer_kwargs"]
-            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"
 
             if fine_tune_trainer_kwargs["use_cpu"]:

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

@@ -317,7 +317,7 @@ 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:
-            logger.info("\tStopping fine-tuning since time_limit is reached")
+            logger.log(15, "Stopping fine-tuning since time_limit is reached")
             control.should_training_stop = True
 
 

autogluon/timeseries/models/presets.py

@@ -145,7 +145,6 @@ def get_default_hps(key):
         },
         "default": {
             "SeasonalNaive": {},
-            "Croston": {},
             "AutoETS": {},
             "NPTS": {},
             "DynamicOptimizedTheta": {},

autogluon/timeseries/version.py

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

{autogluon.timeseries-1.1.2b20241121.dist-info → autogluon.timeseries-1.1.2b20241123.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 1.1.2b20241121
+Version: 1.1.2b20241123
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -35,15 +35,15 @@ Classifier: Topic :: Scientific/Engineering :: Image Recognition
 Requires-Python: >=3.9, <3.13
 Description-Content-Type: text/markdown
 Requires-Dist: joblib<2,>=1.1
-Requires-Dist: numpy<1.29,>=1.21
-Requires-Dist: scipy<1.13,>=1.5.4
+Requires-Dist: numpy<2.1.4,>=1.25.0
+Requires-Dist: scipy<1.16,>=1.5.4
 Requires-Dist: pandas<2.3.0,>=2.0.0
 Requires-Dist: torch<2.6,>=2.2
 Requires-Dist: lightning<2.6,>=2.2
 Requires-Dist: pytorch-lightning
 Requires-Dist: transformers[sentencepiece]<5,>=4.38.0
 Requires-Dist: accelerate<1.0,>=0.32.0
-Requires-Dist: gluonts==0.16.0
+Requires-Dist: gluonts<0.17,>=0.15.0
 Requires-Dist: networkx<4,>=3.0
 Requires-Dist: statsforecast<1.8,>=1.7.0
 Requires-Dist: mlforecast==0.13.4
@@ -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.2b20241121
-Requires-Dist: autogluon.common==1.1.2b20241121
-Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20241121
+Requires-Dist: autogluon.core[raytune]==1.1.2b20241123
+Requires-Dist: autogluon.common==1.1.2b20241123
+Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20241123
 Provides-Extra: all
 Requires-Dist: optimum[onnxruntime]<1.20,>=1.17; extra == "all"
 Provides-Extra: chronos-onnx

{autogluon.timeseries-1.1.2b20241121.dist-info → autogluon.timeseries-1.1.2b20241123.dist-info}/RECORD

@@ -1,11 +1,11 @@
-autogluon.timeseries-1.1.2b20241121-py3.8-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.timeseries-1.1.2b20241123-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=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=5pdrFks0lfiEaMYBllEvk2rSws63pWwiDMSnsCdKbgk,90
+autogluon/timeseries/version.py,sha256=Uw4ihPcz-ROHAaOD4_1aKoi-vhViU2s9HiM3pGEv4Bw,90
 autogluon/timeseries/configs/__init__.py,sha256=BTtHIPCYeGjqgOcvqb8qPD4VNX-ICKOg6wnkew1cPOE,98
 autogluon/timeseries/configs/presets_configs.py,sha256=cLat8ecLlWrI-SC5KLBDCX2SbVXaucemy2pjxJAtSY0,2543
 autogluon/timeseries/dataset/__init__.py,sha256=UvnhAN5tjgxXTHoZMQDy64YMDj4Xxa68yY7NP4vAw0o,81
@@ -16,7 +16,7 @@ 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=cRboVn7MRC6XgHyzL2iZRCF6T-jKU6DPth-x_x_6MRk,12473
+autogluon/timeseries/models/presets.py,sha256=25HfBMhhzO_R-mpQSfBlpKEXOzkS0P7EZh16ipwXANE,12446
 autogluon/timeseries/models/abstract/__init__.py,sha256=wvDsQAZIV0N3AwBeMaGItoQ82trEfnT-nol2AAOIxBg,102
 autogluon/timeseries/models/abstract/abstract_timeseries_model.py,sha256=82lg2odAxzyhM3dkV3Msqv8AeYLILnnk-UvNnYobzFA,30628
 autogluon/timeseries/models/abstract/model_trial.py,sha256=ENPg_7nsdxIvaNM0o0UShZ3x8jFlRmwRc5m0fGPC0TM,3720
@@ -25,12 +25,12 @@ autogluon/timeseries/models/autogluon_tabular/mlforecast.py,sha256=vfWXLdxYlbzjK
 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=PCf32pzxg3zGbP8DMjJC3cgc8GDdt2dhVHEzc18SVrM,29100
+autogluon/timeseries/models/chronos/model.py,sha256=UEw_TGeHmS-_mhGSQevwRJC0TqlNBlmGCWJaiMyhwNU,29582
 autogluon/timeseries/models/chronos/pipeline/__init__.py,sha256=N-YZH9BGBoi99r5cznJe1zEEjwjIg7cOYIHZkKuJq44,247
 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=hw4QbRy81V9Jkndda9p_KgO9CrtYKbNq5kp0r8m0Rek,13133
+autogluon/timeseries/models/chronos/pipeline/utils.py,sha256=dENQLSN6dumLrTGQ6sbJMq45irdDFOoCarAnWpTbLjk,13134
 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
@@ -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.2b20241121.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.timeseries-1.1.2b20241121.dist-info/METADATA,sha256=LzaeFjdcpiA6u7x_i-SbRjpY2_IyBc8XuLv6X1Iiztg,12388
-autogluon.timeseries-1.1.2b20241121.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.timeseries-1.1.2b20241121.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
-autogluon.timeseries-1.1.2b20241121.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.timeseries-1.1.2b20241121.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.timeseries-1.1.2b20241121.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.timeseries-1.1.2b20241121.dist-info/RECORD,,
+autogluon.timeseries-1.1.2b20241123.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.timeseries-1.1.2b20241123.dist-info/METADATA,sha256=OJFO-uQFZvp1tDBuehznT3edZsmHvkb7NDSgGYhAdxA,12397
+autogluon.timeseries-1.1.2b20241123.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.timeseries-1.1.2b20241123.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.timeseries-1.1.2b20241123.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-1.1.2b20241123.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-1.1.2b20241123.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.timeseries-1.1.2b20241123.dist-info/RECORD,,

{autogluon.timeseries-1.1.2b20241121.dist-info → autogluon.timeseries-1.1.2b20241123.dist-info}/WHEEL

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