autogluon.timeseries 1.0.1b20240320__tar.gz → 1.0.1b20240322__tar.gz

{autogluon.timeseries-1.0.1b20240320 → autogluon.timeseries-1.0.1b20240322}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 1.0.1b20240320
+Version: 1.0.1b20240322
 Summary: AutoML for Image, Text, and Tabular Data
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -132,4 +132,5 @@ Classifier: Topic :: Scientific/Engineering :: Image Recognition
 Requires-Python: >=3.8, <3.12
 Description-Content-Type: text/markdown
 Provides-Extra: tests
+Provides-Extra: chronos-cpu
 Provides-Extra: all

{autogluon.timeseries-1.0.1b20240320 → autogluon.timeseries-1.0.1b20240322}/setup.py

@@ -28,6 +28,8 @@ install_requires = [
     "torch",  # version range defined in `core/_setup_utils.py`
     "lightning",  # version range defined in `core/_setup_utils.py`
     "pytorch_lightning",  # version range defined in `core/_setup_utils.py`
+    "transformers[sentencepiece]",  # version range defined in `core/_setup_utils.py`
+    "accelerate",  # version range defined in `core/_setup_utils.py`
     "statsmodels>=0.13.0,<0.15",
     "gluonts>=0.14.0,<0.15",
     "networkx",  # version range defined in `core/_setup_utils.py`
@@ -53,9 +55,12 @@ extras_require = {
         "isort>=5.10",
         "black~=23.0",
     ],
+    "chronos-cpu": [  # for faster CPU inference in pretrained models
+        "optimum[onnxruntime,openvino,nncf]>=1.17,<1.18",
+    ],
 }
 
-extras_require["all"] = []
+extras_require["all"] = list(set.union(*(set(extras_require[extra]) for extra in ["chronos-cpu"])))
 
 install_requires = ag.get_dependency_version_ranges(install_requires)
 

{autogluon.timeseries-1.0.1b20240320 → autogluon.timeseries-1.0.1b20240322}/src/autogluon/timeseries/version.py

@@ -1,3 +1,3 @@
 """This is the autogluon version file."""
-__version__ = '1.0.1b20240320'
+__version__ = '1.0.1b20240322'
 __lite__ = False

{autogluon.timeseries-1.0.1b20240320 → autogluon.timeseries-1.0.1b20240322}/src/autogluon.timeseries.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 1.0.1b20240320
+Version: 1.0.1b20240322
 Summary: AutoML for Image, Text, and Tabular Data
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -132,4 +132,5 @@ Classifier: Topic :: Scientific/Engineering :: Image Recognition
 Requires-Python: >=3.8, <3.12
 Description-Content-Type: text/markdown
 Provides-Extra: tests
+Provides-Extra: chronos-cpu
 Provides-Extra: all

autogluon.timeseries-1.0.1b20240322/src/autogluon.timeseries.egg-info/requires.txt

@@ -0,0 +1,35 @@
+joblib<2,>=1.1
+numpy<1.29,>=1.21
+scipy<1.13,>=1.5.4
+pandas<2.2.0,>=2.0.0
+torch<2.2,>=2.1
+lightning<2.2,>=2.1
+pytorch_lightning<2.2,>=2.1
+transformers[sentencepiece]<4.39.0,>=4.38.0
+accelerate<0.22.0,>=0.21.0
+statsmodels<0.15,>=0.13.0
+gluonts<0.15,>=0.14.0
+networkx<4,>=3.0
+statsforecast<1.5,>=1.4.0
+mlforecast<0.10.1,>=0.10.0
+utilsforecast<0.0.11,>=0.0.10
+tqdm<5,>=4.38
+orjson~=3.9
+tensorboard<3,>=2.9
+autogluon.core[raytune]==1.0.1b20240322
+autogluon.common==1.0.1b20240322
+autogluon.tabular[catboost,lightgbm,xgboost]==1.0.1b20240322
+
+[all]
+optimum[nncf,onnxruntime,openvino]<1.18,>=1.17
+
+[chronos-cpu]
+optimum[nncf,onnxruntime,openvino]<1.18,>=1.17
+
+[tests]
+pytest
+ruff>=0.0.285
+flaky<4,>=3.7
+pytest-timeout<3,>=2.1
+isort>=5.10
+black~=23.0

autogluon.timeseries-1.0.1b20240320/src/autogluon.timeseries.egg-info/requires.txt

@@ -1,29 +0,0 @@
-joblib<2,>=1.1
-numpy<1.29,>=1.21
-scipy<1.13,>=1.5.4
-pandas<2.2.0,>=2.0.0
-torch<2.2,>=2.0
-lightning<2.2,>=2.0.0
-pytorch_lightning<2.2,>=2.0.0
-statsmodels<0.15,>=0.13.0
-gluonts<0.15,>=0.14.0
-networkx<4,>=3.0
-statsforecast<1.5,>=1.4.0
-mlforecast<0.10.1,>=0.10.0
-utilsforecast<0.0.11,>=0.0.10
-tqdm<5,>=4.38
-orjson~=3.9
-tensorboard<3,>=2.9
-autogluon.core[raytune]==1.0.1b20240320
-autogluon.common==1.0.1b20240320
-autogluon.tabular[catboost,lightgbm,xgboost]==1.0.1b20240320
-
-[all]
-
-[tests]
-pytest
-ruff>=0.0.285
-flaky<4,>=3.7
-pytest-timeout<3,>=2.1
-isort>=5.10
-black~=23.0