autogluon.timeseries 1.2.1b20250224__py3-none-any.whl → 1.4.1b20251215__py3-none-any.whl

{autogluon.timeseries-1.2.1b20250224.dist-info → autogluon_timeseries-1.4.1b20251215.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: autogluon.timeseries
-Version: 1.2.1b20250224
+Version: 1.4.1b20251215
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -9,7 +9,6 @@ Project-URL: Documentation, https://auto.gluon.ai
 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: Intended Audience :: Education
 Classifier: Intended Audience :: Developers
@@ -24,51 +23,64 @@ Classifier: Operating System :: Microsoft :: Windows
 Classifier: Operating System :: POSIX
 Classifier: Operating System :: Unix
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Scientific/Engineering :: Information Analysis
 Classifier: Topic :: Scientific/Engineering :: Image Recognition
-Requires-Python: >=3.9, <3.13
+Requires-Python: >=3.10, <3.14
 Description-Content-Type: text/markdown
-License-File: ../LICENSE
-License-File: ../NOTICE
-Requires-Dist: joblib<2,>=1.1
-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.34.0
+License-File: LICENSE
+License-File: NOTICE
+Requires-Dist: joblib<1.7,>=1.2
+Requires-Dist: numpy<2.4.0,>=1.25.0
+Requires-Dist: scipy<1.17,>=1.5.4
+Requires-Dist: pandas<2.4.0,>=2.0.0
+Requires-Dist: torch<2.10,>=2.6
+Requires-Dist: lightning<2.6,>=2.5.1
+Requires-Dist: transformers[sentencepiece]<4.58,>=4.51.0
+Requires-Dist: accelerate<2.0,>=0.34.0
 Requires-Dist: gluonts<0.17,>=0.15.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: statsforecast<2.0.1,>=1.7.0
-Requires-Dist: mlforecast<0.14,>0.13
-Requires-Dist: utilsforecast<0.2.11,>=0.2.3
-Requires-Dist: coreforecast<0.0.16,>=0.0.12
+Requires-Dist: statsforecast<2.0.2,>=1.7.0
+Requires-Dist: mlforecast<0.15.0,>=0.14.0
+Requires-Dist: utilsforecast<0.2.12,>=0.2.3
+Requires-Dist: coreforecast<0.0.17,>=0.0.12
 Requires-Dist: fugue>=0.9.0
 Requires-Dist: tqdm<5,>=4.38
 Requires-Dist: orjson~=3.9
+Requires-Dist: einops<1,>=0.7
+Requires-Dist: chronos-forecasting<2.4,>=2.2.0
+Requires-Dist: peft<0.18,>=0.13.0
 Requires-Dist: tensorboard<3,>=2.9
-Requires-Dist: autogluon.core[raytune]==1.2.1b20250224
-Requires-Dist: autogluon.common==1.2.1b20250224
-Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.2.1b20250224
-Provides-Extra: all
-Provides-Extra: chronos-onnx
-Requires-Dist: optimum[onnxruntime]<1.20,>=1.17; extra == "chronos-onnx"
-Provides-Extra: chronos-openvino
-Requires-Dist: optimum-intel[nncf,openvino]<1.17,>=1.15; extra == "chronos-openvino"
-Requires-Dist: optimum[nncf,openvino]<1.19,>=1.17; extra == "chronos-openvino"
+Requires-Dist: autogluon.core==1.4.1b20251215
+Requires-Dist: autogluon.common==1.4.1b20251215
+Requires-Dist: autogluon.features==1.4.1b20251215
+Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.4.1b20251215
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"
 Requires-Dist: ruff>=0.0.285; extra == "tests"
 Requires-Dist: flaky<4,>=3.7; extra == "tests"
 Requires-Dist: pytest-timeout<3,>=2.1; extra == "tests"
+Provides-Extra: ray
+Requires-Dist: autogluon.core[raytune]==1.4.1b20251215; extra == "ray"
+Provides-Extra: all
+Requires-Dist: autogluon.core[raytune]==1.4.1b20251215; extra == "all"
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 
 
@@ -79,7 +91,7 @@ Requires-Dist: pytest-timeout<3,>=2.1; extra == "tests"
 
 [![Latest Release](https://img.shields.io/github/v/release/autogluon/autogluon)](https://github.com/autogluon/autogluon/releases)
 [![Conda Forge](https://img.shields.io/conda/vn/conda-forge/autogluon.svg)](https://anaconda.org/conda-forge/autogluon)
-[![Python Versions](https://img.shields.io/badge/python-3.9%20%7C%203.10%20%7C%203.11%20%7C%203.12-blue)](https://pypi.org/project/autogluon/)
+[![Python Versions](https://img.shields.io/badge/python-3.10%20%7C%203.11%20%7C%203.12%20%7C%203.13-blue)](https://pypi.org/project/autogluon/)
 [![Downloads](https://pepy.tech/badge/autogluon/month)](https://pepy.tech/project/autogluon)
 [![GitHub license](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](./LICENSE)
 [![Discord](https://img.shields.io/discord/1043248669505368144?color=7289da&label=Discord&logo=discord&logoColor=ffffff)](https://discord.gg/wjUmjqAc2N)
@@ -96,7 +108,7 @@ AutoGluon, developed by AWS AI, automates machine learning tasks enabling you to
 
 ## 💾 Installation
 
-AutoGluon is supported on Python 3.9 - 3.12 and is available on Linux, MacOS, and Windows.
+AutoGluon is supported on Python 3.10 - 3.13 and is available on Linux, MacOS, and Windows.
 
 You can install AutoGluon with:
 
@@ -112,15 +124,15 @@ Build accurate end-to-end ML models in just 3 lines of code!
 
 ```python
 from autogluon.tabular import TabularPredictor
-predictor = TabularPredictor(label="class").fit("train.csv")
+predictor = TabularPredictor(label="class").fit("train.csv", presets="best")
 predictions = predictor.predict("test.csv")
 ```
 
 | AutoGluon Task      |                                                                                Quickstart                                                                                |                                                                                API                                                                                |
 |:--------------------|:------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|:-----------------------------------------------------------------------------------------------------------------------------------------------------------------:|
 | TabularPredictor    | [![Quick Start](https://img.shields.io/static/v1?label=&message=tutorial&color=grey)](https://auto.gluon.ai/stable/tutorials/tabular/tabular-quick-start.html) |                 [![API](https://img.shields.io/badge/api-reference-blue.svg)](https://auto.gluon.ai/stable/api/autogluon.tabular.TabularPredictor.html)                 |
-| MultiModalPredictor | [![Quick Start](https://img.shields.io/static/v1?label=&message=tutorial&color=grey)](https://auto.gluon.ai/stable/tutorials/multimodal/multimodal_prediction/multimodal-quick-start.html)            | [![API](https://img.shields.io/badge/api-reference-blue.svg)](https://auto.gluon.ai/stable/api/autogluon.multimodal.MultiModalPredictor.html) |
 | TimeSeriesPredictor | [![Quick Start](https://img.shields.io/static/v1?label=&message=tutorial&color=grey)](https://auto.gluon.ai/stable/tutorials/timeseries/forecasting-quick-start.html)            | [![API](https://img.shields.io/badge/api-reference-blue.svg)](https://auto.gluon.ai/stable/api/autogluon.timeseries.TimeSeriesPredictor.html) |
+| MultiModalPredictor | [![Quick Start](https://img.shields.io/static/v1?label=&message=tutorial&color=grey)](https://auto.gluon.ai/stable/tutorials/multimodal/multimodal_prediction/multimodal-quick-start.html)            | [![API](https://img.shields.io/badge/api-reference-blue.svg)](https://auto.gluon.ai/stable/api/autogluon.multimodal.MultiModalPredictor.html) |
 
 ## :mag: Resources
 
@@ -143,7 +155,10 @@ Below is a curated list of recent tutorials and talks on AutoGluon. A comprehens
 - [Benchmarking Multimodal AutoML for Tabular Data with Text Fields](https://datasets-benchmarks-proceedings.neurips.cc/paper/2021/file/9bf31c7ff062936a96d3c8bd1f8f2ff3-Paper-round2.pdf) (*NeurIPS*, 2021) ([BibTeX](CITING.md#autogluonmultimodal))
 - [XTab: Cross-table Pretraining for Tabular Transformers](https://proceedings.mlr.press/v202/zhu23k/zhu23k.pdf) (*ICML*, 2023)
 - [AutoGluon-TimeSeries: AutoML for Probabilistic Time Series Forecasting](https://arxiv.org/abs/2308.05566) (*AutoML Conf*, 2023) ([BibTeX](CITING.md#autogluontimeseries))
-- [TabRepo: A Large Scale Repository of Tabular Model Evaluations and its AutoML Applications](https://arxiv.org/pdf/2311.02971.pdf) (*Under Review*, 2024)
+- [TabRepo: A Large Scale Repository of Tabular Model Evaluations and its AutoML Applications](https://arxiv.org/pdf/2311.02971.pdf) (*AutoML Conf*, 2024)
+- [AutoGluon-Multimodal (AutoMM): Supercharging Multimodal AutoML with Foundation Models](https://arxiv.org/pdf/2404.16233) (*AutoML Conf*, 2024) ([BibTeX](CITING.md#autogluonmultimodal))
+- [Multi-layer Stack Ensembles for Time Series Forecasting](https://arxiv.org/abs/2511.15350) (*AutoML Conf*, 2025) ([BibTeX](CITING.md#autogluontimeseries))
+- [Chronos-2: From Univariate to Universal Forecasting](https://arxiv.org/abs/2510.15821) (*Arxiv*, 2025) ([BibTeX](CITING.md#autogluontimeseries))
 
 ### Articles
 - [AutoGluon-TimeSeries: Every Time Series Forecasting Model In One Library](https://towardsdatascience.com/autogluon-timeseries-every-time-series-forecasting-model-in-one-library-29a3bf6879db) (*Towards Data Science*, Jan 2024)
@@ -169,5 +184,3 @@ We are actively accepting code contributions to the AutoGluon project. If you ar
 ## :classical_building: License
 
 This library is licensed under the Apache 2.0 License.
-
-

autogluon_timeseries-1.4.1b20251215.dist-info/RECORD

@@ -0,0 +1,103 @@
+autogluon.timeseries-1.4.1b20251215-py3.11-nspkg.pth,sha256=kAlKxjI5mE3Pwwqphu2maN5OBQk8W8ew70e_qbI1c6A,482
+autogluon/timeseries/__init__.py,sha256=_CrLLc1fkjen7UzWoO0Os8WZoHOgvZbHKy46I8v_4k4,304
+autogluon/timeseries/learner.py,sha256=9kGn0ACGfbyZRlZmwkrgBbkwq7c2715yKDCh1EK3EWQ,14961
+autogluon/timeseries/predictor.py,sha256=0jFU63BNGIq0Y33hj-XmXobetYNObuIHpNKk29ydUUA,103144
+autogluon/timeseries/regressor.py,sha256=HDdqi7MYRheW3uZy5c50sqVDAHap0ooyQBdOvKEKkWM,11718
+autogluon/timeseries/splitter.py,sha256=2rypDxDKkqOC2v5nPJ6m0cmHQTZ9D6qUFrQV1HC9lz4,2329
+autogluon/timeseries/version.py,sha256=l3nl_aPmIgPmCpYmL6FcaeflMlmZ3u1TwxKWLeIjjMg,91
+autogluon/timeseries/configs/__init__.py,sha256=wiLBwxZkDTQBJkSJ9-xz3p_yJxX0dbHe108dS1P5O6A,183
+autogluon/timeseries/configs/hyperparameter_presets.py,sha256=uxL5H9k9kiDcXl16bWZ57Y1HUwwwmfSaQEpUrS-J4yU,2018
+autogluon/timeseries/configs/predictor_presets.py,sha256=sYnrs0A1ae5yzHcLLMl4v5oU9kPB3m7nK-rPzjCasFo,3577
+autogluon/timeseries/dataset/__init__.py,sha256=UvnhAN5tjgxXTHoZMQDy64YMDj4Xxa68yY7NP4vAw0o,81
+autogluon/timeseries/dataset/ts_dataframe.py,sha256=IOIkwV_VPV3JvilNt98gZ77gMHIpk-Ug-trDvqSk_Jg,52228
+autogluon/timeseries/metrics/__init__.py,sha256=iFGLMOtDJ470dbmmx1BsdUKBx4RwI6ZQGFat3Z-wpzI,3567
+autogluon/timeseries/metrics/abstract.py,sha256=_A0Ex1Ay91TPDStZ8DBiBMkIyLUusdARbuDiylHJ0yQ,11499
+autogluon/timeseries/metrics/point.py,sha256=K1Fn0_-Ycxz1hYHd-u1X7q9X-Jt7Dp9bNvUHV6RRg7A,18274
+autogluon/timeseries/metrics/quantile.py,sha256=f8SMVt9rV0sY9lk8B1Bjxx219IjajuJjhOSD95p_z24,4602
+autogluon/timeseries/metrics/utils.py,sha256=_Nz6GLbs91WhqN1PoA53wD4xEEuPIQ0juV5l9rDmkFo,970
+autogluon/timeseries/models/__init__.py,sha256=zPdwxiveOTGU9658tDPMFXbflZ5fzd_AJdbCacbfZ0s,1375
+autogluon/timeseries/models/registry.py,sha256=dkuyKG5UK2xiGtXcsuyRDXrI-YC84zkPre8Z3wt9T_A,2115
+autogluon/timeseries/models/abstract/__init__.py,sha256=Htfkjjc3vo92RvyM8rIlQ0PLWt3jcrCKZES07UvCMV0,146
+autogluon/timeseries/models/abstract/abstract_timeseries_model.py,sha256=7j_ULO_d7SUUprHqnMjF_4pz8rDXODyyFeboJaQohAw,32489
+autogluon/timeseries/models/abstract/model_trial.py,sha256=xKD6Nw8hIqAq4HxNVcGUhr9BuEqzFn7FX0TenvZHU0Q,3753
+autogluon/timeseries/models/abstract/tunable.py,sha256=thl_wJjB9ao1T5NNF1RVH5k3yFqmao0irX-eUNqDs8k,7111
+autogluon/timeseries/models/autogluon_tabular/__init__.py,sha256=E5fZsdFPgVdyCVyj5bGmn_lQFlCMn2NvuRLBMcCFvhM,205
+autogluon/timeseries/models/autogluon_tabular/mlforecast.py,sha256=FJlYqMZJaltTlh54LMrDOgICgGanIymBI2F4OevVQ6A,36690
+autogluon/timeseries/models/autogluon_tabular/per_step.py,sha256=kc0OIveCUfMbl1yGANW42EaRFZZNmlr1AJdcG-nqihA,23360
+autogluon/timeseries/models/autogluon_tabular/transforms.py,sha256=AkXEInK4GocApU5GylECH01qgz5cLLLqC9apuN0eUbQ,2873
+autogluon/timeseries/models/autogluon_tabular/utils.py,sha256=Fn3Vu_Q0PCtEUbtNgLp1xIblg7dOdpFlF3W5kLHgruI,63
+autogluon/timeseries/models/chronos/__init__.py,sha256=dIoAImmZc0dTlut4CZkJxcg1bpuHKZkS8x8Y6fBoUAY,113
+autogluon/timeseries/models/chronos/chronos2.py,sha256=86pWzlExDMp6FGD0ob1Tui2cJQBgeYljGTRUqIRzBNM,15177
+autogluon/timeseries/models/chronos/model.py,sha256=49QF2aM-piPLWd2oD1_0qaWTCUaInCxx_8aHBh8Q-wY,33675
+autogluon/timeseries/models/chronos/utils.py,sha256=t80Cz3EdEOzI2youjVSNYrz1_Xhi-BiaiLsodI5fYtM,14446
+autogluon/timeseries/models/ensemble/__init__.py,sha256=3_Vn6RHpjouthrEoXs1guKUpUX6JoUgMVCgxPt2pyLw,1302
+autogluon/timeseries/models/ensemble/abstract.py,sha256=gAaspq4f67MTfs7KW6ADVU0KfPeBKySPstCqUeC7JYs,4579
+autogluon/timeseries/models/ensemble/ensemble_selection.py,sha256=hepycVJTtbibzTKq5Sk04L_vUuYlLFItkSybaCc_Jv8,6366
+autogluon/timeseries/models/ensemble/per_item_greedy.py,sha256=UlPtSBwzbzVtcf_o8HGMJDbpnWb95KcuvcS_z-AiT5k,7868
+autogluon/timeseries/models/ensemble/array_based/__init__.py,sha256=u4vGTH9gP6oATYKkxnvoiDZvc5rqfnfgrODHxIvHP7U,207
+autogluon/timeseries/models/ensemble/array_based/abstract.py,sha256=Oci1XEgFFTle0JF5Z8PhnMjG1iPrhhtunoKUPUPhTLw,10190
+autogluon/timeseries/models/ensemble/array_based/models.py,sha256=Jb_Rw944B1fG4_ZnysgpmQ-rbMUpv9MXfdj6t1xq2yY,8603
+autogluon/timeseries/models/ensemble/array_based/regressor/__init__.py,sha256=OJPZZzowllw7Ks0aXF8Hye1_1Ql8XhRfdtv3e3A_4AE,424
+autogluon/timeseries/models/ensemble/array_based/regressor/abstract.py,sha256=MSeYWwxH1mL3lrsHbDpzAg61Bovs2Fxkxl3qzj5QrXE,2771
+autogluon/timeseries/models/ensemble/array_based/regressor/linear_stacker.py,sha256=4rUYEXcyyZ8hPITzg1tSDWmHSGfwqrTp5dd-b7MP5Hs,7245
+autogluon/timeseries/models/ensemble/array_based/regressor/per_quantile_tabular.py,sha256=GIa2CtP3bl7uN3i4t54WPod4JxIhA9nKIyr7tx9B08E,3763
+autogluon/timeseries/models/ensemble/array_based/regressor/tabular.py,sha256=prH6vSmRu4UBUjIdAHnLF0aH8oxHUA8ciaNP9ou9uyA,4056
+autogluon/timeseries/models/ensemble/weighted/__init__.py,sha256=_LipTsDnYvTFmjZWsb1Vrm-eALsVVfUlF2gOpcaqE2Q,206
+autogluon/timeseries/models/ensemble/weighted/abstract.py,sha256=meGVoSfPOjmEwTKGRTUQJ1N9bZtpewJ217TGqKNye04,1839
+autogluon/timeseries/models/ensemble/weighted/basic.py,sha256=KsFcdmhkjywqSYvx9rdWoFzjLO-czKsOj3CWuC61SS4,3715
+autogluon/timeseries/models/ensemble/weighted/greedy.py,sha256=NN51NnrbHd7zdJs2kErm6bs-B_OcbjlkD2j2Q00IUOE,2589
+autogluon/timeseries/models/gluonts/__init__.py,sha256=YfyNYOkhhNsloA4MAavfmqKO29_q6o4lwPoV7L4_h7M,355
+autogluon/timeseries/models/gluonts/abstract.py,sha256=qJ60DSkzSI4E1kx5RGeGBehkiMvcAVGSUXYSpZXo8nk,27699
+autogluon/timeseries/models/gluonts/dataset.py,sha256=ApR-r4o0OV4jQ2hYUppJ4yjvWX02JoHod5O4acEKiHw,5074
+autogluon/timeseries/models/gluonts/models.py,sha256=1Z3x3-jVoae5X4cSnDIgJMvTJ9_O94aDSW8HEnBaL5k,25907
+autogluon/timeseries/models/local/__init__.py,sha256=TiKY7M6Foy8vtshfZiStEH58_XG62w4oF1TQYAQ1B0s,344
+autogluon/timeseries/models/local/abstract_local_model.py,sha256=7pbyE4vhXgoCEcHAhxpxBVCOEG-LSrBptGwjLXd-s8o,11335
+autogluon/timeseries/models/local/naive.py,sha256=w0XuMcgcTvTUEi2iXcd6BGvyHKB-kpqbv9c9iK4pMOA,7490
+autogluon/timeseries/models/local/npts.py,sha256=mKuDsGnaYV8QkIgGR8se-1pXb2JAxzafESt2g_21ENA,4530
+autogluon/timeseries/models/local/statsforecast.py,sha256=gt9evIxlymisBlBZU7aRFtZQ3mgyX7a0xtmvFyKRXK4,33275
+autogluon/timeseries/models/multi_window/__init__.py,sha256=Bq7AT2Jxdd4WNqmjTdzeqgNiwn1NCyWp4tBIWaM-zfI,60
+autogluon/timeseries/models/multi_window/multi_window_model.py,sha256=bv8_ux-7JXPwhbFXeBN893xQo6echCCMwqH4aEMK250,12937
+autogluon/timeseries/models/toto/__init__.py,sha256=rQaVjZJV5ZsJGC0jhQ6CA4nYeXdV1KtlyDz2i2usQnY,54
+autogluon/timeseries/models/toto/dataloader.py,sha256=wUrK3mcSEhaWmxpv3rAqmp1ZbLnXbEP4F77hAT2-VXg,3566
+autogluon/timeseries/models/toto/hf_pretrained_model.py,sha256=E2agvz4jdUFhYEiavLTuBIripbl2KLGgdfr8eZXkqOM,7290
+autogluon/timeseries/models/toto/model.py,sha256=ObMPp_Wn2cccT7osWyIfc15gk-hcWDT38p3r-uSdZmM,9412
+autogluon/timeseries/models/toto/_internal/__init__.py,sha256=tKkiux9bD2Xu0AuVyTEx_sNOZutcluC7-d7tn7wsmec,193
+autogluon/timeseries/models/toto/_internal/dataset.py,sha256=jpKX3LV4FkcGGgUPTzpwdR_7UZEFMfwXIQQZVkQ_I6E,6090
+autogluon/timeseries/models/toto/_internal/forecaster.py,sha256=HhRQwqC6Y_Gr93fT-EpilWFjjxY5zR9GsNPN2JPztN4,18479
+autogluon/timeseries/models/toto/_internal/backbone/__init__.py,sha256=hq5W62boH6HiEP8z3sHkI6_KM-Dd6TkDfWDm6DYE3J8,63
+autogluon/timeseries/models/toto/_internal/backbone/attention.py,sha256=ez7N8ygH4Q1gU88EuoSeF1675JcoAAxocvyF4i0JuGI,9347
+autogluon/timeseries/models/toto/_internal/backbone/backbone.py,sha256=Vy2AHnbRrc68ax41KPf0IP3RkXA7GtTgzIXr6lSAp-w,10079
+autogluon/timeseries/models/toto/_internal/backbone/distribution.py,sha256=8NXiaEVLuvjTW7L1t1RzooZFNERWv50zyLddbAwuYpo,2502
+autogluon/timeseries/models/toto/_internal/backbone/kvcache.py,sha256=QSVCrnbS2oD7wkJodZbP9XMVmrfCH6M3Zp44siF28Fg,5399
+autogluon/timeseries/models/toto/_internal/backbone/rope.py,sha256=UohCHvsOP2Q2g6IXDWXQsYpBZ0JDZ0JjtFq0ZnRCF6g,3389
+autogluon/timeseries/models/toto/_internal/backbone/rotary_embedding_torch.py,sha256=TsdcUpQUQes4dtrWb6citENGrXK8hE3M8DyZ2kslEyE,11488
+autogluon/timeseries/models/toto/_internal/backbone/scaler.py,sha256=NQno9Ycm2wf4tZJneoOtbbyZ-ez0Z5R37XJng9rPn_4,13694
+autogluon/timeseries/models/toto/_internal/backbone/transformer.py,sha256=K7S-fPZZOl65luFMpPQ3LC2QuNN4SunTLDTxp-bZWUc,12364
+autogluon/timeseries/trainer/__init__.py,sha256=_tw3iioJfvtIV7wnjtEMv0yS8oabmCFxDnGRodYE7RI,72
+autogluon/timeseries/trainer/ensemble_composer.py,sha256=zGa8vocPQGsHf-7ti8DsHwjEA176FkCt7up2LwWCK4Y,19465
+autogluon/timeseries/trainer/model_set_builder.py,sha256=kROApbu10_ro-GVYlnx3oTKZj2TcNswWbOFB1QyBCOc,10737
+autogluon/timeseries/trainer/prediction_cache.py,sha256=KKs22UUGrVfQN_81IgzL7Bfc8tjWk3k6YW3uHURaSs0,5496
+autogluon/timeseries/trainer/trainer.py,sha256=igUml9D4eXOq_fryhQquU2tnl8hyrdUTzdqUz0IJ6Ig,56308
+autogluon/timeseries/trainer/utils.py,sha256=7N4vRP6GFUlRAahxQ9PqppdIMFqMz3wpZ5u-_onR24M,588
+autogluon/timeseries/transforms/__init__.py,sha256=fKlT4pkJ_8Gl7IUTc3uSDzt2Xow5iH5w6fPB3ePNrTg,127
+autogluon/timeseries/transforms/covariate_scaler.py,sha256=CpTtokiE1uEg_RJa4kEUUuBwXZpPL11OC2fgCkRpGlQ,6986
+autogluon/timeseries/transforms/target_scaler.py,sha256=sAOohPBaStZx_V8aaaQacDbfEqqWRjYUtDLxdhkRKww,6092
+autogluon/timeseries/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autogluon/timeseries/utils/constants.py,sha256=qjFWoouIQ5nJfx9Fmm4svN191ultb4XWW4NQSHeiGW4,542
+autogluon/timeseries/utils/features.py,sha256=pI5Nu7Pj4ZrNat7vGjEAqqPO-XJGsdlH2rCUMEJLhC8,23527
+autogluon/timeseries/utils/forecast.py,sha256=-w94i4DZaervXAZ_c1M7I4iLrPnVax8yC6pgv46bEjc,2228
+autogluon/timeseries/utils/timer.py,sha256=qDROHYG_Z8fulMpyZMrRhfQoTneazTzYhur4qjqqydA,5799
+autogluon/timeseries/utils/warning_filters.py,sha256=SroNhLU3kwbD8anM58vdxWq36Z8j_uiY42mEt0ya-JI,2589
+autogluon/timeseries/utils/datetime/__init__.py,sha256=bTMR8jLh1LW55vHjbOr1zvWRMF_PqbvxpS-cUcNIDWI,173
+autogluon/timeseries/utils/datetime/base.py,sha256=3NdsH3NDq4cVAOSoy3XpaNixyNlbjy4DJ_YYOGuu9x4,1341
+autogluon/timeseries/utils/datetime/lags.py,sha256=dijskkPDJXhXbRHGQZPhUFuEom3typKbOeET7cxkHGY,5965
+autogluon/timeseries/utils/datetime/seasonality.py,sha256=-w3bULdkIZKP-JrO1ahHLyNCanLhejocHlasZShuwA0,802
+autogluon/timeseries/utils/datetime/time_features.py,sha256=kEOFls4Nzh8nO0Pcz1DwLsC_NA3hMI4JUlZI3kuvuts,2666
+autogluon_timeseries-1.4.1b20251215.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon_timeseries-1.4.1b20251215.dist-info/licenses/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon_timeseries-1.4.1b20251215.dist-info/METADATA,sha256=S_fGdXEpxl3d6ttYaRHXUdt0SjyXQ9pS3x8JAiMYgns,13425
+autogluon_timeseries-1.4.1b20251215.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+autogluon_timeseries-1.4.1b20251215.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon_timeseries-1.4.1b20251215.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon_timeseries-1.4.1b20251215.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon_timeseries-1.4.1b20251215.dist-info/RECORD,,

{autogluon.timeseries-1.2.1b20250224.dist-info → autogluon_timeseries-1.4.1b20251215.dist-info}/WHEEL

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

autogluon/timeseries/configs/presets_configs.py

@@ -1,79 +0,0 @@
-"""Preset configurations for autogluon.timeseries Predictors"""
-
-from autogluon.timeseries.models.presets import get_default_hps
-
-# TODO: change default HPO settings when other HPO strategies (e.g., Ray tune) are available
-# TODO: add refit_full arguments once refitting is available
-
-TIMESERIES_PRESETS_CONFIGS = dict(
-    best_quality={"hyperparameters": "default", "num_val_windows": 2},
-    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,
-    },
-    chronos_mini={
-        "hyperparameters": {"Chronos": {"model_path": "mini"}},
-        "skip_model_selection": True,
-    },
-    chronos_small={
-        "hyperparameters": {"Chronos": {"model_path": "small"}},
-        "skip_model_selection": True,
-    },
-    chronos_base={
-        "hyperparameters": {"Chronos": {"model_path": "base"}},
-        "skip_model_selection": True,
-    },
-    chronos_large={
-        "hyperparameters": {"Chronos": {"model_path": "large", "batch_size": 8}},
-        "skip_model_selection": True,
-    },
-    chronos_ensemble={
-        "hyperparameters": {
-            "Chronos": {"model_path": "small"},
-            **get_default_hps("light_inference"),
-        }
-    },
-    chronos_large_ensemble={
-        "hyperparameters": {
-            "Chronos": {"model_path": "large", "batch_size": 8},
-            **get_default_hps("light_inference"),
-        }
-    },
-)
-
-TIMESERIES_PRESETS_ALIASES = dict(
-    chronos="chronos_small",
-    best="best_quality",
-    high="high_quality",
-    medium="medium_quality",
-    bq="best_quality",
-    hq="high_quality",
-    mq="medium_quality",
-)
-
-# update with aliases
-TIMESERIES_PRESETS_CONFIGS = {
-    **TIMESERIES_PRESETS_CONFIGS,
-    **{k: TIMESERIES_PRESETS_CONFIGS[v].copy() for k, v in TIMESERIES_PRESETS_ALIASES.items()},
-}

autogluon/timeseries/evaluator.py

@@ -1,6 +0,0 @@
-class TimeSeriesEvaluator:
-    def __init__(self, *args, **kwargs):
-        raise ValueError(
-            "`TimeSeriesEvaluator` has been deprecated. "
-            "Please use the metrics defined in `autogluon.timeseries.metrics` instead."
-        )

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

@@ -1,11 +0,0 @@
-from .chronos import ChronosPipeline
-from .chronos_bolt import ChronosBoltPipeline
-from .base import BaseChronosPipeline, ForecastType
-
-
-__all__ = [
-    "BaseChronosPipeline",
-    "ChronosBoltPipeline",
-    "ChronosPipeline",
-    "ForecastType",
-]

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

@@ -1,160 +0,0 @@
-# Authors: Lorenzo Stella <stellalo@amazon.com>, Caner Turkmen <atturkm@amazon.com>
-
-from enum import Enum
-from pathlib import Path
-from typing import TYPE_CHECKING, Dict, List, Optional, Tuple, Union
-
-import torch
-
-from .utils import left_pad_and_stack_1D
-
-if TYPE_CHECKING:
-    from transformers import PreTrainedModel
-
-
-class ForecastType(Enum):
-    SAMPLES = "samples"
-    QUANTILES = "quantiles"
-
-
-class PipelineRegistry(type):
-    REGISTRY: Dict[str, "PipelineRegistry"] = {}
-
-    def __new__(cls, name, bases, attrs):
-        """See, https://github.com/faif/python-patterns."""
-        new_cls = type.__new__(cls, name, bases, attrs)
-        if name is not None:
-            cls.REGISTRY[name] = new_cls
-        if aliases := attrs.get("_aliases"):
-            for alias in aliases:
-                cls.REGISTRY[alias] = new_cls
-        return new_cls
-
-
-class BaseChronosPipeline(metaclass=PipelineRegistry):
-    forecast_type: ForecastType
-    dtypes = {
-        "bfloat16": torch.bfloat16,
-        "float32": torch.float32,
-        "float64": torch.float64,
-    }
-
-    def __init__(self, inner_model: "PreTrainedModel"):
-        """
-        Parameters
-        ----------
-        inner_model : PreTrainedModel
-            A hugging-face transformers PreTrainedModel, e.g., T5ForConditionalGeneration
-        """
-        # for easy access to the inner HF-style model
-        self.inner_model = inner_model
-
-    def _prepare_and_validate_context(self, context: Union[torch.Tensor, List[torch.Tensor]]):
-        if isinstance(context, list):
-            context = left_pad_and_stack_1D(context)
-        assert isinstance(context, torch.Tensor)
-        if context.ndim == 1:
-            context = context.unsqueeze(0)
-        assert context.ndim == 2
-
-        return context
-
-    def predict(
-        self,
-        context: Union[torch.Tensor, List[torch.Tensor]],
-        prediction_length: Optional[int] = None,
-        **kwargs,
-    ):
-        """
-        Get forecasts for the given time series.
-
-        Parameters
-        ----------
-        context
-            Input series. This is either a 1D tensor, or a list
-            of 1D tensors, or a 2D tensor whose first dimension
-            is batch. In the latter case, use left-padding with
-            ``torch.nan`` to align series of different lengths.
-        prediction_length
-            Time steps to predict. Defaults to a model-dependent
-            value if not given.
-
-        Returns
-        -------
-        forecasts
-            Tensor containing forecasts. The layout and meaning
-            of the forecasts values depends on ``self.forecast_type``.
-        """
-        raise NotImplementedError()
-
-    def predict_quantiles(
-        self, context: torch.Tensor, prediction_length: int, quantile_levels: List[float], **kwargs
-    ) -> Tuple[torch.Tensor, torch.Tensor]:
-        """
-        Get quantile and mean forecasts for given time series. All
-        predictions are returned on the CPU.
-
-        Parameters
-        ----------
-        context
-            Input series. This is either a 1D tensor, or a list
-            of 1D tensors, or a 2D tensor whose first dimension
-            is batch. In the latter case, use left-padding with
-            ``torch.nan`` to align series of different lengths.
-        prediction_length
-            Time steps to predict. Defaults to a model-dependent
-            value if not given.
-        quantile_levels: List[float]
-            Quantile levels to compute
-
-        Returns
-        -------
-        quantiles
-            Tensor containing quantile forecasts. Shape
-            (batch_size, prediction_length, num_quantiles)
-        mean
-            Tensor containing mean (point) forecasts. Shape
-            (batch_size, prediction_length)
-        """
-        raise NotImplementedError()
-
-    @classmethod
-    def from_pretrained(
-        cls,
-        pretrained_model_name_or_path: Union[str, Path],
-        *model_args,
-        force=False,
-        **kwargs,
-    ):
-        """
-        Load the model, either from a local path or from the HuggingFace Hub.
-        Supports the same arguments as ``AutoConfig`` and ``AutoModel``
-        from ``transformers``.
-
-        When a local path is provided, supports both a folder or a .tar.gz archive.
-        """
-        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
-
-            local_model_path = cache_model_from_s3(str(pretrained_model_name_or_path), force=force)
-            return cls.from_pretrained(local_model_path, *model_args, **kwargs)
-
-        torch_dtype = kwargs.get("torch_dtype", "auto")
-        if torch_dtype != "auto" and isinstance(torch_dtype, str):
-            kwargs["torch_dtype"] = cls.dtypes[torch_dtype]
-
-        config = AutoConfig.from_pretrained(pretrained_model_name_or_path, **kwargs)
-        is_valid_config = hasattr(config, "chronos_pipeline_class") or hasattr(config, "chronos_config")
-
-        if not is_valid_config:
-            raise ValueError("Not a Chronos config file")
-
-        pipeline_class_name = getattr(config, "chronos_pipeline_class", "ChronosPipeline")
-        class_: Optional[BaseChronosPipeline] = PipelineRegistry.REGISTRY.get(pipeline_class_name)  # type: ignore
-        if class_ is None:
-            raise ValueError(f"Trying to load unknown pipeline class: {pipeline_class_name}")
-
-        return class_.from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)