autogluon.timeseries 1.1.2b20240628__tar.gz → 1.1.2b20240630__tar.gz

{autogluon.timeseries-1.1.2b20240628 → autogluon.timeseries-1.1.2b20240630}/PKG-INFO

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

{autogluon.timeseries-1.1.2b20240628 → autogluon.timeseries-1.1.2b20240630}/setup.py

@@ -32,10 +32,10 @@ install_requires = [
     "accelerate",  # version range defined in `core/_setup_utils.py`
     "gluonts==0.15.1",
     "networkx",  # version range defined in `core/_setup_utils.py`
-    # TODO: update statsforecast to v1.5.0 - resolve antlr4-python3-runtime dependency clash with multimodal
-    "statsforecast>=1.4.0,<1.5",
+    "statsforecast>=1.7.0,<1.8",
     "mlforecast>=0.10.0,<0.10.1",
-    "utilsforecast>=0.0.10,<0.0.11",
+    "utilsforecast<=0.1.9",  # to prevent breaking changes that propagate through mlforecast's dependency
+    "fugue>=0.9.0",  # prevent dependency clash with omegaconf
     "tqdm",  # version range defined in `core/_setup_utils.py`
     "orjson~=3.9",  # use faster JSON implementation in GluonTS
     # TODO v1.1: use lightning[pytorch-extra] instead of explicitly installing tensorboard

{autogluon.timeseries-1.1.2b20240628 → autogluon.timeseries-1.1.2b20240630}/src/autogluon/timeseries/models/local/statsforecast.py

@@ -204,8 +204,6 @@ class ARIMAModel(AbstractProbabilisticStatsForecastModel):
         This significantly speeds up fitting and usually leads to no change in accuracy.
     """
 
-    # TODO: This model requires statsforecast >= 1.5.0, so it will only be available after we upgrade the dependency
-
     allowed_local_model_args = [
         "order",
         "seasonal_order",
@@ -491,8 +489,7 @@ class AbstractConformalizedStatsForecastModel(AbstractStatsForecastModel):
         return pd.DataFrame(predictions)
 
 
-# TODO: Starting from StatsForecast v1.5.0, AutoCES can inherit from AbstractProbabilisticStatsForecastModel
-class AutoCESModel(AbstractConformalizedStatsForecastModel):
+class AutoCESModel(AbstractProbabilisticStatsForecastModel):
     """Forecasting with an Complex Exponential Smoothing model where the model selection is performed using the
     Akaike Information Criterion.
 

{autogluon.timeseries-1.1.2b20240628 → autogluon.timeseries-1.1.2b20240630}/src/autogluon/timeseries/models/presets.py

@@ -10,6 +10,7 @@ from autogluon.timeseries.metrics import TimeSeriesScorer
 
 from . import (
     ADIDAModel,
+    ARIMAModel,
     AutoARIMAModel,
     AutoCESModel,
     AutoETSModel,
@@ -38,9 +39,6 @@ from . import (
 from .abstract import AbstractTimeSeriesModel
 from .multi_window.multi_window_model import MultiWindowBacktestingModel
 
-# TODO: Enable ARIMAModel after upgrading to StatsForecast >=1.5.0 - currently ARIMA model is broken
-
-
 logger = logging.getLogger(__name__)
 
 ModelHyperparameters = Dict[str, Any]
@@ -68,6 +66,7 @@ MODEL_TYPES = dict(
     NPTS=NPTSModel,
     Theta=ThetaModel,
     ETS=ETSModel,
+    ARIMA=ARIMAModel,
     ADIDA=ADIDAModel,
     CrostonSBA=CrostonSBAModel,
     IMAPA=IMAPAModel,

{autogluon.timeseries-1.1.2b20240628 → autogluon.timeseries-1.1.2b20240630}/src/autogluon/timeseries/version.py

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

{autogluon.timeseries-1.1.2b20240628 → autogluon.timeseries-1.1.2b20240630}/src/autogluon.timeseries.egg-info/PKG-INFO

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

{autogluon.timeseries-1.1.2b20240628 → autogluon.timeseries-1.1.2b20240630}/src/autogluon.timeseries.egg-info/requires.txt

@@ -9,15 +9,16 @@ transformers[sentencepiece]<4.41.0,>=4.38.0
 accelerate<0.22.0,>=0.21.0
 gluonts==0.15.1
 networkx<4,>=3.0
-statsforecast<1.5,>=1.4.0
+statsforecast<1.8,>=1.7.0
 mlforecast<0.10.1,>=0.10.0
-utilsforecast<0.0.11,>=0.0.10
+utilsforecast<=0.1.9
+fugue>=0.9.0
 tqdm<5,>=4.38
 orjson~=3.9
 tensorboard<3,>=2.9
-autogluon.core[raytune]==1.1.2b20240628
-autogluon.common==1.1.2b20240628
-autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20240628
+autogluon.core[raytune]==1.1.2b20240630
+autogluon.common==1.1.2b20240630
+autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20240630
 
 [all]
 optimum[onnxruntime]<1.19,>=1.17