autogluon.timeseries 1.1.1b20240516__tar.gz → 1.1.1b20240518__tar.gz

{autogluon.timeseries-1.1.1b20240516 → autogluon.timeseries-1.1.1b20240518}/PKG-INFO

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

{autogluon.timeseries-1.1.1b20240516 → autogluon.timeseries-1.1.1b20240518}/src/autogluon/timeseries/models/chronos/model.py

@@ -16,27 +16,27 @@ logger = logging.getLogger(__name__)
 
 # allowed HuggingFace model paths with custom parameter definitions
 MODEL_CONFIGS = {
-    "amazon/chronos-t5-tiny": {
+    "chronos-t5-tiny": {
         "num_gpus": 0,  # minimum number of required GPUs
         "default_torch_dtype": "auto",
         "default_batch_size": 16,
     },
-    "amazon/chronos-t5-mini": {
+    "chronos-t5-mini": {
         "num_gpus": 0,
         "default_torch_dtype": "auto",
         "default_batch_size": 16,
     },
-    "amazon/chronos-t5-small": {
+    "chronos-t5-small": {
         "num_gpus": 1,
         "default_torch_dtype": "bfloat16",
         "default_batch_size": 16,
     },
-    "amazon/chronos-t5-base": {
+    "chronos-t5-base": {
         "num_gpus": 1,
         "default_torch_dtype": "bfloat16",
         "default_batch_size": 16,
     },
-    "amazon/chronos-t5-large": {
+    "chronos-t5-large": {
         "num_gpus": 1,
         "default_torch_dtype": "bfloat16",
         "default_batch_size": 8,
@@ -45,11 +45,11 @@ MODEL_CONFIGS = {
 
 
 MODEL_ALIASES = {
-    "tiny": "amazon/chronos-t5-tiny",
-    "mini": "amazon/chronos-t5-mini",
-    "small": "amazon/chronos-t5-small",
-    "base": "amazon/chronos-t5-base",
-    "large": "amazon/chronos-t5-large",
+    "tiny": "autogluon/chronos-t5-tiny",
+    "mini": "autogluon/chronos-t5-mini",
+    "small": "autogluon/chronos-t5-small",
+    "base": "autogluon/chronos-t5-base",
+    "large": "autogluon/chronos-t5-large",
 }
 
 
@@ -75,10 +75,10 @@ class ChronosModel(AbstractTimeSeriesModel):
 
     Other Parameters
     ----------------
-    model_path: str, default = "amazon/chronos-t5-small"
+    model_path: str, default = "autogluon/chronos-t5-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., ``amazon/chronos-t5-{model_size}``) can be specified with aliases
+        Chronos models (i.e., ``autogluon/chronos-t5-{model_size}``) can be specified with aliases
         ``tiny``, ``mini`` , ``small``, ``base``, and ``large``.
     batch_size : int, default = 16
         Size of batches used during inference
@@ -106,7 +106,7 @@ class ChronosModel(AbstractTimeSeriesModel):
 
     # default number of samples for prediction
     default_num_samples: int = 20
-    default_model_path = "amazon/chronos-t5-small"
+    default_model_path = "autogluon/chronos-t5-small"
     maximum_context_length = 512
 
     def __init__(
@@ -185,9 +185,10 @@ class ChronosModel(AbstractTimeSeriesModel):
     @property
     def ag_default_config(self) -> Dict[str, Any]:
         """The default configuration of the model used by AutoGluon if the model is one of those
-        defined in MODEL_CONFIGS. For now, these are ``amazon/chronos-t5-*`` family of models.
+        defined in MODEL_CONFIGS. For now, these are ``autogluon/chronos-t5-*`` family of models.
         """
-        return MODEL_CONFIGS.get(self.model_path, {})
+        model_name = str(self.model_path).split("/")[-1]
+        return MODEL_CONFIGS.get(model_name, {})
 
     @property
     def min_num_gpus(self) -> int:

{autogluon.timeseries-1.1.1b20240516 → autogluon.timeseries-1.1.1b20240518}/src/autogluon/timeseries/models/chronos/pipeline.py

@@ -163,7 +163,7 @@ class MeanScaleUniformBins(ChronosTokenizer):
     def output_transform(self, samples: torch.Tensor, scale: torch.Tensor) -> torch.Tensor:
         scale_unsqueezed = scale.unsqueeze(-1).unsqueeze(-1)
         indices = torch.clamp(
-            samples - self.config.n_special_tokens,
+            samples - self.config.n_special_tokens - 1,
             min=0,
             max=len(self.centers) - 1,
         )

{autogluon.timeseries-1.1.1b20240516 → autogluon.timeseries-1.1.1b20240518}/src/autogluon/timeseries/predictor.py

@@ -146,7 +146,7 @@ class TimeSeriesPredictor(TimeSeriesPredictorDeprecatedMixin):
     """
 
     predictor_file_name = "predictor.pkl"
-    _predictor_version_file_name = "__version__"
+    _predictor_version_file_name = "version.txt"
     _predictor_log_file_name = "predictor_log.txt"
 
     def __init__(
@@ -1041,8 +1041,27 @@ class TimeSeriesPredictor(TimeSeriesPredictorDeprecatedMixin):
 
     @classmethod
     def _load_version_file(cls, path: str) -> str:
+        """
+        Loads the version file that is part of the saved predictor artifact.
+
+        Parameters
+        ----------
+        path: str
+            The path that would be used to load the predictor via `predictor.load(path)`
+
+        Returns
+        -------
+        The version of AutoGluon used to fit the predictor, as a string.
+
+        """
         version_file_path = os.path.join(path, cls._predictor_version_file_name)
-        version = load_str.load(path=version_file_path)
+        try:
+            version = load_str.load(path=version_file_path)
+        except:
+            # Loads the old version file used in `autogluon.timeseries<=1.1.0`, named `__version__`.
+            # This file name was changed because Kaggle does not allow uploading files named `__version__`.
+            version_file_path = os.path.join(path, "__version__")
+            version = load_str.load(path=version_file_path)
         return version
 
     @classmethod
@@ -1077,7 +1096,7 @@ class TimeSeriesPredictor(TimeSeriesPredictorDeprecatedMixin):
         except:
             logger.warning(
                 f'WARNING: Could not find version file at "{os.path.join(path, cls._predictor_version_file_name)}".\n'
-                f"This means that the predictor was fit in a version `<=0.7.0`."
+                f"This means that the predictor was fit in an AutoGluon version `<=0.7.0`."
             )
             version_saved = "Unknown (Likely <=0.7.0)"
 

{autogluon.timeseries-1.1.1b20240516 → autogluon.timeseries-1.1.1b20240518}/src/autogluon/timeseries/trainer/abstract_trainer.py

@@ -292,6 +292,10 @@ class AbstractTimeSeriesTrainer(SimpleAbstractTrainer):
         self.cache_predictions = cache_predictions
         self.hpo_results = {}
 
+        if self._cached_predictions_path.exists():
+            logger.debug(f"Removing existing cached predictions file {self._cached_predictions_path}")
+            self._cached_predictions_path.unlink()
+
     def save_train_data(self, data: TimeSeriesDataFrame, verbose: bool = True) -> None:
         path = os.path.join(self.path_data, "train.pkl")
         save_pkl.save(path=path, object=data, verbose=verbose)

{autogluon.timeseries-1.1.1b20240516 → autogluon.timeseries-1.1.1b20240518}/src/autogluon/timeseries/version.py

@@ -1,3 +1,3 @@
 """This is the autogluon version file."""
-__version__ = '1.1.1b20240516'
+__version__ = '1.1.1b20240518'
 __lite__ = False

{autogluon.timeseries-1.1.1b20240516 → autogluon.timeseries-1.1.1b20240518}/src/autogluon.timeseries.egg-info/PKG-INFO

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

{autogluon.timeseries-1.1.1b20240516 → autogluon.timeseries-1.1.1b20240518}/src/autogluon.timeseries.egg-info/requires.txt

@@ -15,9 +15,9 @@ utilsforecast<0.0.11,>=0.0.10
 tqdm<5,>=4.38
 orjson~=3.9
 tensorboard<3,>=2.9
-autogluon.core[raytune]==1.1.1b20240516
-autogluon.common==1.1.1b20240516
-autogluon.tabular[catboost,lightgbm,xgboost]==1.1.1b20240516
+autogluon.core[raytune]==1.1.1b20240518
+autogluon.common==1.1.1b20240518
+autogluon.tabular[catboost,lightgbm,xgboost]==1.1.1b20240518
 
 [all]
 optimum[onnxruntime]<1.19,>=1.17