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(),