autogluon.timeseries 1.3.2b20250711__tar.gz → 1.3.2b20250713__tar.gz

{autogluon.timeseries-1.3.2b20250711 → autogluon.timeseries-1.3.2b20250713}/PKG-INFO

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

{autogluon.timeseries-1.3.2b20250711 → autogluon.timeseries-1.3.2b20250713}/setup.py

@@ -21,7 +21,7 @@ version = ag.update_version(version)
 submodule = "timeseries"
 install_requires = [
     # version ranges added in ag.get_dependency_version_ranges()
-    "joblib>=1.1,<2",
+    "joblib",  # version range defined in `core/_setup_utils.py`
     "numpy",  # version range defined in `core/_setup_utils.py`
     "scipy",  # version range defined in `core/_setup_utils.py`
     "pandas",  # version range defined in `core/_setup_utils.py`

{autogluon.timeseries-1.3.2b20250711 → autogluon.timeseries-1.3.2b20250713}/src/autogluon/timeseries/models/autogluon_tabular/mlforecast.py

@@ -11,6 +11,7 @@ from sklearn.base import BaseEstimator
 
 import autogluon.core as ag
 from autogluon.core.models import AbstractModel as AbstractTabularModel
+from autogluon.features import AutoMLPipelineFeatureGenerator
 from autogluon.tabular.registry import ag_model_registry
 from autogluon.timeseries.dataset.ts_dataframe import ITEMID, TIMESTAMP, TimeSeriesDataFrame
 from autogluon.timeseries.metrics.abstract import TimeSeriesScorer
@@ -35,14 +36,18 @@ class TabularModel(BaseEstimator):
     def __init__(self, model_class: Type[AbstractTabularModel], model_kwargs: Optional[dict] = None):
         self.model_class = model_class
         self.model_kwargs = {} if model_kwargs is None else model_kwargs
+        self.feature_pipeline = AutoMLPipelineFeatureGenerator()
 
-    def fit(self, *args, **kwargs):
+    def fit(self, X: pd.DataFrame, y: pd.Series, X_val: pd.DataFrame, y_val: pd.Series, **kwargs):
         self.model = self.model_class(**self.model_kwargs)
-        self.model.fit(*args, **kwargs)
+        X = self.feature_pipeline.fit_transform(X=X)
+        X_val = self.feature_pipeline.transform(X=X_val)
+        self.model.fit(X=X, y=y, X_val=X_val, y_val=y_val, **kwargs)
         return self
 
-    def predict(self, *args, **kwargs):
-        return self.model.predict(*args, **kwargs)
+    def predict(self, X: pd.DataFrame, **kwargs):
+        X = self.feature_pipeline.transform(X=X)
+        return self.model.predict(X=X, **kwargs)
 
     def get_params(self, deep=True):
         params = {"model_class": self.model_class, "model_kwargs": self.model_kwargs}
@@ -346,7 +351,7 @@ class AbstractMLForecastModel(AbstractTimeSeriesModel):
             max_num_samples=model_params["max_num_samples"],
         )
 
-        with set_loggers_level(regex=r"^autogluon.tabular.*", level=logging.ERROR):
+        with set_loggers_level(regex=r"^autogluon\.(tabular|features).*", level=logging.ERROR):
             tabular_model = self._create_tabular_model(
                 model_name=model_params["model_name"], model_hyperparameters=model_params["model_hyperparameters"]
             )
@@ -364,12 +369,12 @@ class AbstractMLForecastModel(AbstractTimeSeriesModel):
 
         self._save_residuals_std(val_df)
 
-    def get_tabular_model(self) -> AbstractTabularModel:
-        """Get the unerlyin tabular regression model."""
+    def get_tabular_model(self) -> TabularModel:
+        """Get the underlying tabular regression model."""
         assert "mean" in self._mlf.models_, "Call `fit` before calling `get_tabular_model`"
         mean_estimator = self._mlf.models_["mean"]
         assert isinstance(mean_estimator, TabularModel)
-        return mean_estimator.model
+        return mean_estimator
 
     def _save_residuals_std(self, val_df: pd.DataFrame) -> None:
         """Compute standard deviation of residuals for each item using the validation set.

{autogluon.timeseries-1.3.2b20250711 → autogluon.timeseries-1.3.2b20250713}/src/autogluon/timeseries/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.3.2b20250711"
+__version__ = "1.3.2b20250713"
 __lite__ = False

{autogluon.timeseries-1.3.2b20250711 → autogluon.timeseries-1.3.2b20250713}/src/autogluon.timeseries.egg-info/PKG-INFO

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

{autogluon.timeseries-1.3.2b20250711 → autogluon.timeseries-1.3.2b20250713}/src/autogluon.timeseries.egg-info/requires.txt

@@ -1,4 +1,4 @@
-joblib<2,>=1.1
+joblib<1.7,>=1.2
 numpy<2.4.0,>=1.25.0
 scipy<1.17,>=1.5.4
 pandas<2.4.0,>=2.0.0
@@ -17,10 +17,10 @@ fugue>=0.9.0
 tqdm<5,>=4.38
 orjson~=3.9
 tensorboard<3,>=2.9
-autogluon.core[raytune]==1.3.2b20250711
-autogluon.common==1.3.2b20250711
-autogluon.features==1.3.2b20250711
-autogluon.tabular[catboost,lightgbm,xgboost]==1.3.2b20250711
+autogluon.core[raytune]==1.3.2b20250713
+autogluon.common==1.3.2b20250713
+autogluon.features==1.3.2b20250713
+autogluon.tabular[catboost,lightgbm,xgboost]==1.3.2b20250713
 
 [all]