diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 085d05ab..03974a9c 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -20,3 +20,4 @@ repos: hooks: - id: mypy args: [--ignore-missing-imports] + exclude: 'setup.py' diff --git a/mlforecast/__init__.py b/mlforecast/__init__.py index 1c07e24d..e8ba600a 100644 --- a/mlforecast/__init__.py +++ b/mlforecast/__init__.py @@ -1,3 +1,3 @@ -__version__ = "0.13.0" +__version__ = "0.13.1" __all__ = ['MLForecast'] from mlforecast.forecast import MLForecast diff --git a/nbs/docs/getting-started/install.ipynb b/nbs/docs/getting-started/install.ipynb index 70d19e70..1dff39e3 100644 --- a/nbs/docs/getting-started/install.ipynb +++ b/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", diff --git a/settings.ini b/settings.ini index b18817a7..0cda5e6a 100644 --- a/settings.ini +++ b/settings.ini @@ -8,21 +8,22 @@ author = José Morales author_email = jmoralz92@gmail.com 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 diff --git a/setup.py b/setup.py index a18fc95b..0d38f493 100644 --- a/setup.py +++ b/setup.py @@ -31,6 +31,7 @@ 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, @@ -38,6 +39,7 @@ 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(),