autogluon.timeseries 1.1.2b20241124__py3-none-any.whl → 1.2__py3-none-any.whl

autogluon/timeseries/models/chronos/model.py

@@ -133,10 +133,13 @@ class ChronosModel(AbstractTimeSeriesModel):
         and may truncate the context further. For example, original Chronos models have a context length of 512, but
         Chronos-Bolt models handle contexts up to 2048.
     optimization_strategy : {None, "onnx", "openvino"}, default = None
-        Optimization strategy to use for inference on CPUs. If None, the model will use the default implementation.
+        [deprecated] Optimization strategy to use for inference on CPUs. If None, the model will use the default implementation.
         If `onnx`, the model will be converted to ONNX and the inference will be performed using ONNX. If ``openvino``,
         inference will be performed with the model compiled to OpenVINO. These optimizations are only available for
-        the original set of Chronos models, and not in Chronos-Bolt where they are not needed.
+        the original set of Chronos models, and not in Chronos-Bolt where they are not needed. You will need to
+        install the appropriate dependencies `optimum[onnxruntime]` or `optimum[openvino,nncf] optimum-intel[openvino,nncf]`
+        for optimizations to work. Note that support for optimization strategies is deprecated, and will be removed
+        in a future release. We recommend using Chronos-Bolt models for fast inference on the CPU.
     torch_dtype : torch.dtype or {"auto", "bfloat16", "float32", "float64"}, default = "auto"
         Torch data type for model weights, provided to ``from_pretrained`` method of Hugging Face AutoModels. If
         original Chronos models are specified and the model size is ``small``, ``base``, or ``large``, the
@@ -202,6 +205,15 @@ class ChronosModel(AbstractTimeSeriesModel):
         self.optimization_strategy: Optional[Literal["onnx", "openvino"]] = hyperparameters.get(
             "optimization_strategy", None
         )
+        if self.optimization_strategy is not None:
+            warnings.warn(
+                (
+                    "optimization_strategy is deprecated and will be removed in a future release. "
+                    "We recommend using Chronos-Bolt models for fast inference on the CPU."
+                ),
+                category=FutureWarning,
+                stacklevel=3,
+            )
         self.context_length = hyperparameters.get("context_length")
 
         if self.context_length is not None and self.context_length > self.maximum_context_length:

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

@@ -556,7 +556,9 @@ class ChronosPipeline(BaseChronosPipeline):
                     from optimum.onnxruntime import ORTModelForSeq2SeqLM
                 except ImportError:
                     raise ImportError(
-                        "Huggingface Optimum library must be installed with ONNX for using the `onnx` strategy"
+                        "Huggingface Optimum library must be installed with ONNX for using the `onnx` strategy. "
+                        "Please try running `pip install optimum[onnxruntime]` or use Chronos-Bolt models for "
+                        "faster performance on the CPU."
                     )
 
                 assert kwargs.pop("device_map", "cpu") in ["cpu", "auto"], "ONNX mode only available on the CPU"
@@ -567,7 +569,9 @@ class ChronosPipeline(BaseChronosPipeline):
                     from optimum.intel import OVModelForSeq2SeqLM
                 except ImportError:
                     raise ImportError(
-                        "Huggingface Optimum library must be installed with OpenVINO for using the `openvino` strategy"
+                        "Huggingface Optimum library must be installed with OpenVINO for using the `openvino` strategy. "
+                        "Please try running `pip install optimum-intel[openvino,nncf] optimum[openvino,nncf]` or use "
+                        "Chronos-Bolt models for faster performance on the CPU."
                     )
                 with set_loggers_level(regex=r"^optimum.*", level=logging.ERROR):
                     inner_model = OVModelForSeq2SeqLM.from_pretrained(

autogluon/timeseries/predictor.py

@@ -447,9 +447,9 @@ class TimeSeriesPredictor(TimeSeriesPredictorDeprecatedMixin):
 
             If ``train_data`` contains covariates or static features, they will be interpreted as follows:
 
-                * columns with ``int``, ``bool`` and ``float`` dtypes are interpreted as continuous (real-valued) features
-                * columns with ``object``, ``str`` and ``category`` dtypes are as interpreted as categorical features
-                * columns with other dtypes are ignored
+            * columns with ``int``, ``bool`` and ``float`` dtypes are interpreted as continuous (real-valued) features
+            * columns with ``object``, ``str`` and ``category`` dtypes are as interpreted as categorical features
+            * columns with other dtypes are ignored
 
             To ensure that the column type is interpreted correctly, please convert it to one of the above dtypes.
             For example, to ensure that column "store_id" with dtype ``int`` is interpreted as a category, change
@@ -501,13 +501,18 @@ class TimeSeriesPredictor(TimeSeriesPredictorDeprecatedMixin):
 
             - ``"fast_training"``: fit simple statistical models (``ETS``, ``Theta``, ``Naive``, ``SeasonalNaive``) + fast tree-based models ``RecursiveTabular``
               and ``DirectTabular``. These models are fast to train but may not be very accurate.
-            - ``"medium_quality"``: all models mentioned above + deep learning model ``TemporalFusionTransformer``. Default setting that produces good forecasts
+            - ``"medium_quality"``: all models mentioned above + deep learning model ``TemporalFusionTransformer`` + Chronos-Bolt (small). Default setting that produces good forecasts
               with reasonable training time.
-            - ``"high_quality"``: All ML models available in AutoGluon + additional statistical models (``NPTS``, ``AutoETS``, ``AutoARIMA``, ``Croston``,
+            - ``"high_quality"``: All ML models available in AutoGluon + additional statistical models (``NPTS``, ``AutoETS``,
               ``DynamicOptimizedTheta``). Much more accurate than ``medium_quality``, but takes longer to train.
             - ``"best_quality"``: Same models as in ``"high_quality"``, but performs validation with multiple backtests. Usually better than ``high_quality``, but takes even longer to train.
 
-            Available presets with the `Chronos <https://github.com/amazon-science/chronos-forecasting>`_ model:
+            Available presets with the new, faster `Chronos-Bolt <https://github.com/amazon-science/chronos-forecasting>`_ model:
+
+            - ``"bolt_{model_size}"``: where model size is one of ``tiny,mini,small,base``. Uses the Chronos-Bolt pretrained model for zero-shot forecasting.
+              See the documentation for ``ChronosModel`` or see `Hugging Face <https://huggingface.co/collections/amazon/chronos-models-65f1791d630a8d57cb718444>`_ for more information.
+
+            Available presets with the original `Chronos <https://github.com/amazon-science/chronos-forecasting>`_ model:
 
             - ``"chronos_{model_size}"``: where model size is one of ``tiny,mini,small,base,large``. Uses the Chronos pretrained model for zero-shot forecasting.
               See the documentation for ``ChronosModel`` or see `Hugging Face <https://huggingface.co/collections/amazon/chronos-models-65f1791d630a8d57cb718444>`_ for more information.

autogluon/timeseries/version.py

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

{autogluon.timeseries-1.1.2b20241124.dist-info → autogluon.timeseries-1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 1.1.2b20241124
+Version: 1.2
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -10,7 +10,7 @@ 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: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Education
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -42,7 +42,7 @@ 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: accelerate<1.0,>=0.34.0
 Requires-Dist: gluonts<0.17,>=0.15.0
 Requires-Dist: networkx<4,>=3.0
 Requires-Dist: statsforecast<1.8,>=1.7.0
@@ -53,11 +53,10 @@ 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.2b20241124
-Requires-Dist: autogluon.common==1.1.2b20241124
-Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20241124
+Requires-Dist: autogluon.core[raytune]==1.2
+Requires-Dist: autogluon.common==1.2
+Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.2
 Provides-Extra: all
-Requires-Dist: optimum[onnxruntime]<1.20,>=1.17; extra == "all"
 Provides-Extra: chronos-onnx
 Requires-Dist: optimum[onnxruntime]<1.20,>=1.17; extra == "chronos-onnx"
 Provides-Extra: chronos-openvino

{autogluon.timeseries-1.1.2b20241124.dist-info → autogluon.timeseries-1.2.dist-info}/RECORD

@@ -1,11 +1,11 @@
-autogluon.timeseries-1.1.2b20241124-py3.8-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.timeseries-1.2-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/predictor.py,sha256=W9RhTUS_WFMAR7BXiHIYblKHCvGlY8WRTfJJ2E0pBn0,85471
 autogluon/timeseries/regressor.py,sha256=tqQ2zWImxpuEyaAM0DeCjOZ-xcWUYZbCXsqd471xXxQ,8351
 autogluon/timeseries/splitter.py,sha256=eghGwAAN2_cxGk5aJBILgjGWtLzjxJcytMy49gg_q18,3061
-autogluon/timeseries/version.py,sha256=a-gYeszw4OmNwMxFD5nO4ov38d4ZpGM3fypUKRVe5gU,90
+autogluon/timeseries/version.py,sha256=OeRnWY70kLYV8NjtUUPZV97_wrZuueMLQ-kr_7xIq-M,79
 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
@@ -25,10 +25,10 @@ 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=UEw_TGeHmS-_mhGSQevwRJC0TqlNBlmGCWJaiMyhwNU,29582
+autogluon/timeseries/models/chronos/model.py,sha256=jYnUdSWMrxdIWhysy-fEVd5w0Z06dQsIYdkh8alwgR0,30343
 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.py,sha256=QmC0ZSHaF85dwAAzbg2elUX10iEhob6gp0Wctfsfq1k,22531
 autogluon/timeseries/models/chronos/pipeline/chronos_bolt.py,sha256=2MJuik-YFgONZ3X2DciAph5So6ABys5ppQhBC81gLyk,20083
 autogluon/timeseries/models/chronos/pipeline/utils.py,sha256=dENQLSN6dumLrTGQ6sbJMq45irdDFOoCarAnWpTbLjk,13134
 autogluon/timeseries/models/ensemble/__init__.py,sha256=kFr11Gmt7lQJu9Rr8HuIPphQN5l1TsoorfbJm_O3a_s,128
@@ -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.2b20241124.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.timeseries-1.1.2b20241124.dist-info/METADATA,sha256=OPAnALM7bzJMgzrzMuuopXtdGb5CGUGZwtF-UqAhgh8,12397
-autogluon.timeseries-1.1.2b20241124.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.timeseries-1.1.2b20241124.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.timeseries-1.1.2b20241124.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.timeseries-1.1.2b20241124.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.timeseries-1.1.2b20241124.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.timeseries-1.1.2b20241124.dist-info/RECORD,,
+autogluon.timeseries-1.2.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.timeseries-1.2.dist-info/METADATA,sha256=G_MApgfVdqFkx3SBXQ8iJ4keIZ4rFSw03Y9MG5mFcPs,12302
+autogluon.timeseries-1.2.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.timeseries-1.2.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.timeseries-1.2.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-1.2.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-1.2.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.timeseries-1.2.dist-info/RECORD,,