autogluon.timeseries 1.0.0b20231122__py3-none-any.whl → 1.0.0b20231124__py3-none-any.whl

autogluon/timeseries/models/autogluon_tabular/mlforecast.py

@@ -296,10 +296,7 @@ class AbstractMLForecastModel(AbstractTimeSeriesModel):
 
     def _get_scale_per_item(self, item_ids: pd.Index) -> pd.Series:
         """Extract the '_scale' values from the scaler object, if available."""
-        if self._scaler is not None:
-            return self._scaler.stats_["_scale"].copy().reindex(item_ids)
-        else:
-            return pd.Series(1.0, index=item_ids)
+        raise NotImplementedError
 
     def _remove_short_ts_and_generate_fallback_forecast(
         self,
@@ -469,7 +466,9 @@ class DirectTabularModel(AbstractMLForecastModel):
 
         raw_predictions = self._mlf.models_["mean"].predict(df)
         predictions = self._postprocess_predictions(raw_predictions, repeated_item_ids=df[MLF_ITEMID])
-        predictions[[MLF_ITEMID, MLF_TIMESTAMP]] = df[[MLF_ITEMID, MLF_TIMESTAMP]].values
+        # Paste columns one by one to preserve dtypes
+        predictions[MLF_ITEMID] = df[MLF_ITEMID].values
+        predictions[MLF_TIMESTAMP] = df[MLF_TIMESTAMP].values
 
         if hasattr(self._mlf.ts, "target_transforms"):
             # Ensure that transforms are fitted only on past data
@@ -498,6 +497,10 @@ class DirectTabularModel(AbstractMLForecastModel):
         column_order = ["mean"] + [col for col in predictions.columns if col != "mean"]
         return predictions[column_order]
 
+    def _get_scale_per_item(self, item_ids: pd.Index) -> pd.Series:
+        # Rescaling is applied in the inverse_transform step, no need to scale predictions
+        return pd.Series(1.0, index=item_ids)
+
     def _get_extra_tabular_init_kwargs(self) -> dict:
         if self.is_quantile_model:
             return {
@@ -603,3 +606,9 @@ class RecursiveTabularModel(AbstractMLForecastModel):
             "problem_type": ag.constants.REGRESSION,
             "eval_metric": self.eval_metric.equivalent_tabular_regression_metric or "mean_absolute_error",
         }
+
+    def _get_scale_per_item(self, item_ids: pd.Index) -> pd.Series:
+        if self._scaler is not None:
+            return self._scaler.stats_["_scale"].copy().reindex(item_ids)
+        else:
+            return pd.Series(1.0, index=item_ids)

autogluon/timeseries/version.py

@@ -1,3 +1,3 @@
 """This is the autogluon version file."""
-__version__ = '1.0.0b20231122'
+__version__ = '1.0.0b20231124'
 __lite__ = False

{autogluon.timeseries-1.0.0b20231122.dist-info → autogluon.timeseries-1.0.0b20231124.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 1.0.0b20231122
+Version: 1.0.0b20231124
 Summary: AutoML for Image, Text, and Tabular Data
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -49,9 +49,10 @@ Requires-Dist: mlforecast <0.10.1,>=0.10.0
 Requires-Dist: utilsforecast <0.0.11,>=0.0.10
 Requires-Dist: tqdm <5,>=4.38
 Requires-Dist: orjson ~=3.9
-Requires-Dist: autogluon.core[raytune] ==1.0.0b20231122
-Requires-Dist: autogluon.common ==1.0.0b20231122
-Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost] ==1.0.0b20231122
+Requires-Dist: tensorboard <3,>=2.9
+Requires-Dist: autogluon.core[raytune] ==1.0.0b20231124
+Requires-Dist: autogluon.common ==1.0.0b20231124
+Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost] ==1.0.0b20231124
 Provides-Extra: all
 Provides-Extra: tests
 Requires-Dist: pytest ; extra == 'tests'

{autogluon.timeseries-1.0.0b20231122.dist-info → autogluon.timeseries-1.0.0b20231124.dist-info}/RECORD

@@ -1,10 +1,10 @@
-autogluon.timeseries-1.0.0b20231122-py3.8-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.timeseries-1.0.0b20231124-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=HVfsoWTG3dXBCc7JbPfHCCYCMwL3zlrqHwLBG33MTJ8,9633
 autogluon/timeseries/predictor.py,sha256=sohEmnK0Z-sf7zhQRR6i7zTtuTigs0QXQrzhxKx8v9o,59016
 autogluon/timeseries/splitter.py,sha256=eghGwAAN2_cxGk5aJBILgjGWtLzjxJcytMy49gg_q18,3061
-autogluon/timeseries/version.py,sha256=Rdh18YbY6uBsX_CKLjMj7xiq_G7Jnakp4KzAO5Y-4uQ,90
+autogluon/timeseries/version.py,sha256=-k59F7BtYG5KzVCW8NlMl325YMkn2027VY6iivBRmI4,90
 autogluon/timeseries/configs/__init__.py,sha256=BTtHIPCYeGjqgOcvqb8qPD4VNX-ICKOg6wnkew1cPOE,98
 autogluon/timeseries/configs/presets_configs.py,sha256=1u6tbOKJdIRULYDu41dlJwXRNswWsjBDF0aR2YhyMQs,479
 autogluon/timeseries/dataset/__init__.py,sha256=UvnhAN5tjgxXTHoZMQDy64YMDj4Xxa68yY7NP4vAw0o,81
@@ -20,7 +20,7 @@ autogluon/timeseries/models/abstract/__init__.py,sha256=wvDsQAZIV0N3AwBeMaGItoQ8
 autogluon/timeseries/models/abstract/abstract_timeseries_model.py,sha256=knKaYsx_sn_KpKgzZhiTCmaTG_W7TAumgnpGXUwzAOU,21841
 autogluon/timeseries/models/abstract/model_trial.py,sha256=_5Nrk4CrG3u35tTd3elekfdnQI2Pn3P9AGS5CE6nuyg,3749
 autogluon/timeseries/models/autogluon_tabular/__init__.py,sha256=r9i6jWcyeLHYClkcMSKRVsfrkBUMxpDrTATNTBc_qgQ,136
-autogluon/timeseries/models/autogluon_tabular/mlforecast.py,sha256=F9hehdIuskalbyTFIxobCQGYSuGbL5aE7--pCCMHUmY,29440
+autogluon/timeseries/models/autogluon_tabular/mlforecast.py,sha256=HNol4LnuElP1a9WBbHjIg4ycHS3tho1f7t7MCgk6v3g,29831
 autogluon/timeseries/models/autogluon_tabular/utils.py,sha256=4-gTrBtizxeMVQlsuscugPqw9unaXWXhS1TVVssfzYY,2125
 autogluon/timeseries/models/ensemble/__init__.py,sha256=kFr11Gmt7lQJu9Rr8HuIPphQN5l1TsoorfbJm_O3a_s,128
 autogluon/timeseries/models/ensemble/abstract_timeseries_ensemble.py,sha256=tifETwmiEGt-YtQ9eNK7ojJ3fBvtFMUJvisbfkIJ7gw,3393
@@ -48,11 +48,11 @@ autogluon/timeseries/utils/datetime/base.py,sha256=MsqIHY14m3QMjSwwtE7Uo1oNwepWU
 autogluon/timeseries/utils/datetime/lags.py,sha256=kcU4liKbHj7KP2ajNU-KLZ8OYSU35EgT4kJjZNSw0Zg,5875
 autogluon/timeseries/utils/datetime/seasonality.py,sha256=kgK_ukw2wCviEB7CZXRVC5HZpBJZu9IsRrvCJ9E_rOE,755
 autogluon/timeseries/utils/datetime/time_features.py,sha256=pROkYyxETQ8rHKfPGhf2paB73C7rWJ2Ui0cCswLqbBg,2562
-autogluon.timeseries-1.0.0b20231122.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.timeseries-1.0.0b20231122.dist-info/METADATA,sha256=7E145kwFYFRjBAOz7iQgUpQvdJGgqSJxTlMUlQ8MGMw,13288
-autogluon.timeseries-1.0.0b20231122.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.timeseries-1.0.0b20231122.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
-autogluon.timeseries-1.0.0b20231122.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.timeseries-1.0.0b20231122.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.timeseries-1.0.0b20231122.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.timeseries-1.0.0b20231122.dist-info/RECORD,,
+autogluon.timeseries-1.0.0b20231124.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.timeseries-1.0.0b20231124.dist-info/METADATA,sha256=S1f5aKQr741Y-QzUCMNAH6W1PEp5e7LNRSvz3g6SKbQ,13324
+autogluon.timeseries-1.0.0b20231124.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.timeseries-1.0.0b20231124.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
+autogluon.timeseries-1.0.0b20231124.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-1.0.0b20231124.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-1.0.0b20231124.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.timeseries-1.0.0b20231124.dist-info/RECORD,,