add polars extra (#368)

.pre-commit-config.yaml

@@ -20,3 +20,4 @@ repos:
     hooks:
       - id: mypy
         args: [--ignore-missing-imports]
+        exclude: 'setup.py'

mlforecast/__init__.py

@@ -1,3 +1,3 @@
-__version__ = "0.13.0"
+__version__ = "0.13.1"
 __all__ = ['MLForecast']
 from mlforecast.forecast import MLForecast

nbs/docs/getting-started/install.ipynb

@@ -71,6 +71,16 @@
     "#### Extras"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "7b575d6e-8ffa-42b1-9b1a-f3a6a28085e0",
+   "metadata": {},
+   "source": [
+    "##### polars\n",
+    "\n",
+    "Using polars dataframes: `pip install \"mlforecast[polars]\"`"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "738f3c23-7969-423b-8177-e08849c9c16f",

settings.ini

@@ -8,21 +8,22 @@ author = José Morales
 author_email = [email protected]
 copyright = Nixtla
 branch = main
-version = 0.13.0
+version = 0.13.1
 min_python = 3.8
 audience = Developers
 language = English
 custom_sidebar = True
 license = apache2
 status = 3
-requirements = cloudpickle coreforecast>=0.0.7 fsspec numba optuna packaging pandas scikit-learn utilsforecast>=0.0.27 window-ops
+requirements = cloudpickle coreforecast>=0.0.10 fsspec numba optuna packaging pandas scikit-learn utilsforecast>=0.1.9 window-ops
 dask_requirements = fugue dask[complete] lightgbm xgboost
 ray_requirements = fugue[ray] lightgbm_ray xgboost_ray
 spark_requirements = fugue pyspark>=3.3 lightgbm xgboost
 aws_requirements = fsspec[s3]
 gcp_requirements = fsspec[gcs]
 azure_requirements = fsspec[adl]
-dev_requirements = black datasetsforecast matplotlib mypy nbdev polars[numpy]>=0.0.0rc0 pre-commit prophet pyarrow ruff shap statsmodels
+polars_requirements = polars[numpy]
+dev_requirements = black datasetsforecast matplotlib mypy nbdev pre-commit prophet pyarrow ruff shap statsmodels
 nbs_path = nbs
 doc_path = _docs
 recursive = True

setup.py

@@ -31,13 +31,15 @@
 aws_requirements = cfg['aws_requirements'].split()
 azure_requirements = cfg['azure_requirements'].split()
 gcp_requirements = cfg['gcp_requirements'].split()
+polars_requirements = cfg['polars_requirements'].split()
 all_extras = [
     dask_requirements,
     ray_requirements,
     spark_requirements,
     aws_requirements,
     azure_requirements,
     gcp_requirements,
+    polars_requirements,
 ]
 dev_requirements = sorted(set(sum([cfg['dev_requirements'].split()] + all_extras, [])))
 min_python = cfg['min_python']
@@ -63,6 +65,7 @@
         'aws': aws_requirements,
         'azure': azure_requirements,
         'gcp': gcp_requirements,
+        'polars': polars_requirements,
         'dev': dev_requirements,
     },
     dependency_links = cfg.get('dep_links','').split(),