autogluon.timeseries 1.1.2b20240819__tar.gz → 1.1.2b20240821__tar.gz

{autogluon.timeseries-1.1.2b20240819 → autogluon.timeseries-1.1.2b20240821}/PKG-INFO

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

{autogluon.timeseries-1.1.2b20240819 → autogluon.timeseries-1.1.2b20240821}/setup.py

@@ -51,8 +51,6 @@ extras_require = {
         "ruff>=0.0.285",
         "flaky>=3.7,<4",
         "pytest-timeout>=2.1,<3",
-        "isort>=5.10",
-        "black~=23.0",
     ],
     "chronos-openvino": [  # for faster CPU inference in pretrained models with OpenVINO
         "optimum-intel[openvino,nncf]>=1.15,<1.17",

{autogluon.timeseries-1.1.2b20240819 → autogluon.timeseries-1.1.2b20240821}/src/autogluon/timeseries/models/local/abstract_local_model.py

@@ -222,18 +222,22 @@ def seasonal_naive_forecast(
 ) -> pd.DataFrame:
     """Generate seasonal naive forecast, predicting the last observed value from the same period."""
 
-    def numpy_ffill(arr: np.ndarray) -> np.ndarray:
-        """Fast implementation of forward fill in numpy."""
+    def numpy_fillna(arr: np.ndarray) -> np.ndarray:
+        """Fast implementation of forward fill + avg fill in numpy."""
+        # First apply forward fill
         idx = np.arange(len(arr))
         mask = np.isnan(arr)
         idx[mask] = 0
-        return arr[np.maximum.accumulate(idx)]
+        arr_filled = arr[np.maximum.accumulate(idx)]
+        # Leading NaNs are filled with the mean
+        arr_filled[np.isnan(arr_filled)] = np.nanmean(arr_filled)
+        return arr_filled
 
     forecast = {}
     # At least seasonal_period + 2 values are required to compute sigma for seasonal naive
     if len(target) > seasonal_period + 1 and seasonal_period > 1:
         if np.isnan(target[-(seasonal_period + 2) :]).any():
-            target = numpy_ffill(target)
+            target = numpy_fillna(target)
 
         indices = [len(target) - seasonal_period + k % seasonal_period for k in range(prediction_length)]
         forecast["mean"] = target[indices]

{autogluon.timeseries-1.1.2b20240819 → autogluon.timeseries-1.1.2b20240821}/src/autogluon/timeseries/version.py

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

{autogluon.timeseries-1.1.2b20240819 → autogluon.timeseries-1.1.2b20240821}/src/autogluon.timeseries.egg-info/PKG-INFO

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

{autogluon.timeseries-1.1.2b20240819 → autogluon.timeseries-1.1.2b20240821}/src/autogluon.timeseries.egg-info/requires.txt

@@ -16,9 +16,9 @@ fugue>=0.9.0
 tqdm<5,>=4.38
 orjson~=3.9
 tensorboard<3,>=2.9
-autogluon.core[raytune]==1.1.2b20240819
-autogluon.common==1.1.2b20240819
-autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20240819
+autogluon.core[raytune]==1.1.2b20240821
+autogluon.common==1.1.2b20240821
+autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20240821
 
 [all]
 optimum[onnxruntime]<1.19,>=1.17
@@ -35,5 +35,3 @@ pytest
 ruff>=0.0.285
 flaky<4,>=3.7
 pytest-timeout<3,>=2.1
-isort>=5.10
-black~=23.0