autogluon.timeseries 1.1.2b20241122__tar.gz → 1.1.2b20241124__tar.gz

{autogluon.timeseries-1.1.2b20241122 → autogluon.timeseries-1.1.2b20241124}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 1.1.2b20241122
+Version: 1.1.2b20241124
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.timeseries-1.1.2b20241122 → autogluon.timeseries-1.1.2b20241124}/src/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
@@ -164,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"
@@ -199,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}."
             )
@@ -342,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,
@@ -435,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-1.1.2b20241122 → autogluon.timeseries-1.1.2b20241124}/src/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-1.1.2b20241122 → autogluon.timeseries-1.1.2b20241124}/src/autogluon/timeseries/models/presets.py

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

{autogluon.timeseries-1.1.2b20241122 → autogluon.timeseries-1.1.2b20241124}/src/autogluon/timeseries/version.py

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

{autogluon.timeseries-1.1.2b20241122 → autogluon.timeseries-1.1.2b20241124}/src/autogluon.timeseries.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 1.1.2b20241122
+Version: 1.1.2b20241124
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.timeseries-1.1.2b20241122 → autogluon.timeseries-1.1.2b20241124}/src/autogluon.timeseries.egg-info/requires.txt

@@ -17,9 +17,9 @@ fugue>=0.9.0
 tqdm<5,>=4.38
 orjson~=3.9
 tensorboard<3,>=2.9
-autogluon.core[raytune]==1.1.2b20241122
-autogluon.common==1.1.2b20241122
-autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20241122
+autogluon.core[raytune]==1.1.2b20241124
+autogluon.common==1.1.2b20241124
+autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20241124
 
 [all]
 optimum[onnxruntime]<1.20,>=1.17