autogluon.timeseries 0.7.0b20230302__tar.gz → 0.7.0b20230303__tar.gz

{autogluon.timeseries-0.7.0b20230302 → autogluon.timeseries-0.7.0b20230303}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 0.7.0b20230302
+Version: 0.7.0b20230303
 Summary: AutoML for Image, Text, and Tabular Data
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.timeseries-0.7.0b20230302 → autogluon.timeseries-0.7.0b20230303}/src/autogluon/timeseries/dataset/ts_dataframe.py

@@ -8,6 +8,7 @@ from typing import Any, Optional, Tuple, Type
 
 import numpy as np
 import pandas as pd
+from joblib.parallel import Parallel, delayed
 from pandas.core.internals import ArrayManager, BlockManager
 
 from autogluon.common.loaders import load_pd
@@ -258,24 +259,27 @@ class TimeSeriesDataFrame(pd.DataFrame):
             raise ValueError(f"all entries in index `{ITEMID}` must be of integer or string dtype")
 
     @classmethod
-    def _construct_pandas_frame_from_iterable_dataset(cls, iterable_dataset: Iterable) -> pd.DataFrame:
-        cls._validate_iterable(iterable_dataset)
-
-        all_ts = []
-        for i, ts in enumerate(iterable_dataset):
+    def _construct_pandas_frame_from_iterable_dataset(
+        cls, iterable_dataset: Iterable, num_cpus: int = -1
+    ) -> pd.DataFrame:
+        def load_single_item(item_id: int, ts: dict) -> pd.DataFrame:
             start_timestamp = ts["start"]
             freq = start_timestamp.freq
             if isinstance(start_timestamp, pd.Period):
                 start_timestamp = start_timestamp.to_timestamp(how="S")
             target = ts["target"]
             datetime_index = tuple(pd.date_range(start_timestamp, periods=len(target), freq=freq))
-            idx = pd.MultiIndex.from_product([(i,), datetime_index], names=[ITEMID, TIMESTAMP])
-            ts_df = pd.Series(target, name="target", index=idx).to_frame()
-            all_ts.append(ts_df)
+            idx = pd.MultiIndex.from_product([(item_id,), datetime_index], names=[ITEMID, TIMESTAMP])
+            return pd.Series(target, name="target", index=idx).to_frame()
+
+        cls._validate_iterable(iterable_dataset)
+        all_ts = Parallel(n_jobs=num_cpus)(
+            delayed(load_single_item)(item_id, ts) for item_id, ts in enumerate(iterable_dataset)
+        )
         return pd.concat(all_ts)
 
     @classmethod
-    def from_iterable_dataset(cls, iterable_dataset: Iterable) -> pd.DataFrame:
+    def from_iterable_dataset(cls, iterable_dataset: Iterable, num_cpus: int = -1) -> pd.DataFrame:
         """Construct a ``TimeSeriesDataFrame`` from an Iterable of dictionaries each of which
         represent a single time series.
 
@@ -294,13 +298,15 @@ class TimeSeriesDataFrame(pd.DataFrame):
                     {"target": [3, 4, 5], "start": pd.Timestamp("01-01-2019", freq='D')},
                     {"target": [6, 7, 8], "start": pd.Timestamp("01-01-2019", freq='D')}
                 ]
+        num_cpus : int, default = -1
+            Number of CPU cores used to process the iterable dataset in parallel. Set to -1 to use all cores.
 
         Returns
         -------
         ts_df: TimeSeriesDataFrame
             A data frame in TimeSeriesDataFrame format.
         """
-        return cls(cls._construct_pandas_frame_from_iterable_dataset(iterable_dataset))
+        return cls(cls._construct_pandas_frame_from_iterable_dataset(iterable_dataset, num_cpus=num_cpus))
 
     @classmethod
     def _load_data_frame_from_file(cls, path: str) -> pd.DataFrame:

{autogluon.timeseries-0.7.0b20230302 → autogluon.timeseries-0.7.0b20230303}/src/autogluon/timeseries/version.py

@@ -1,3 +1,3 @@
 """This is the autogluon version file."""
-__version__ = '0.7.0b20230302'
+__version__ = '0.7.0b20230303'
 __lite__ = False

{autogluon.timeseries-0.7.0b20230302 → autogluon.timeseries-0.7.0b20230303}/src/autogluon.timeseries.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 0.7.0b20230302
+Version: 0.7.0b20230303
 Summary: AutoML for Image, Text, and Tabular Data
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.timeseries-0.7.0b20230302 → autogluon.timeseries-0.7.0b20230303}/src/autogluon.timeseries.egg-info/requires.txt

@@ -10,13 +10,13 @@ networkx<3.0,>=2.3
 statsforecast<1.5,>=1.4.0
 tqdm<5,>=4.38
 ujson<6,>=5
-autogluon.core[raytune]==0.7.0b20230302
-autogluon.common==0.7.0b20230302
-autogluon.tabular[catboost,lightgbm,xgboost]==0.7.0b20230302
+autogluon.core[raytune]==0.7.0b20230303
+autogluon.common==0.7.0b20230303
+autogluon.tabular[catboost,lightgbm,xgboost]==0.7.0b20230303
 
 [all]
-tbats<2,>=1.1
 sktime<0.16,>=0.14
+tbats<2,>=1.1
 pmdarima<1.9,>=1.8.2
 
 [sktime]