pyproject.toml

[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "darts"
version = "0.43.0"
description = "A python library for easy manipulation and forecasting of time series."
readme = "README.md"
requires-python = ">=3.10"
license = "Apache-2.0"
keywords = ["time series", "forecasting"]
maintainers = [
    {name = "Unit8 SA", email = "darts@unit8.co"}
]
classifiers = [
    "Intended Audience :: Science/Research",
    "Intended Audience :: Developers",
    "Programming Language :: Python",
    "Topic :: Software Development",
    "Topic :: Scientific/Engineering",
    "Operating System :: Microsoft :: Windows",
    "Operating System :: POSIX",
    "Operating System :: Unix",
    "Operating System :: MacOS",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: Implementation :: PyPy",
]
dependencies = [
    "holidays>=0.11.1",
    "joblib>=0.16.0",
    "matplotlib>=3.3.0",
    "narwhals>=1.25.1",
    "nfoursid>=1.0.0",
    "numpy>=2.2.0",
    "pandas>=2.2.0",
    "pyod>=0.9.5",
    "requests>=2.22.0",
    "scikit-learn>=1.6.0",
    "scipy>=1.3.2",
    "shap>=0.40.0",
    "statsmodels>=0.14.0",
    "tqdm>=4.60.0",
    "typing-extensions>=4.0.0",
    "xarray>=0.17.0",
]

[project.urls]
Homepage = "https://unit8co.github.io/darts/"
Documentation = "https://unit8co.github.io/darts/"
Repository = "https://github.com/unit8co/darts"
"Bug Tracker" = "https://github.com/unit8co/darts/issues"

[project.optional-dependencies]
# Main feature sets - pip installable with: pip install "darts[torch]"
torch = [
    "torch>=2.0.0",
    "pytorch-lightning>=2.0.0,<2.5.3",
    "tensorboardx>=2.1",
    "huggingface-hub>=0.35.3",
    "safetensors>=0.6.2",
]
notorch = [
    "catboost>=1.0.6",
    "lightgbm>=3.2.0",
    "prophet>=1.1.1",
    "statsforecast>=1.4",
    "xgboost>=2.1.4",
]
all = ["darts[torch,notorch]"]

[tool.uv.sources]
# Use PyTorch CPU-only index when torch-cpu group is enabled
torch = { index = "pytorch-cpu", group = "torch-cpu" }

[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true  # Only use this index for packages explicitly configured to use it

[tool.setuptools]
zip-safe = false

[tool.setuptools.packages.find]
exclude = ["examples*", "datasets*", "docs*", "darts.tests*"]

[tool.setuptools.package-data]
darts = ["py.typed"]

[tool.pytest.ini_options]
addopts = [
    "--strict-markers",
    "--color=yes"
]
markers = [
    "slow: marks tests as slow (deselect with `-m 'not slow'`)",
]


[tool.ruff]
target-version = "py310"
line-length = 88

[tool.ruff.format]
preview = true

[tool.ruff.lint]
select = [
    "E",
    "W", # see: https://pypi.org/project/pycodestyle
    "F", # see: https://pypi.org/project/pyflakes
    "I", #see: https://pypi.org/project/isort/
    "UP", # see: https://docs.astral.sh/ruff/rules/#pyupgrade-up
#    "D", # see: https://pypi.org/project/pydocstyle
]
ignore = [
    "E203",
    "E402",  # todo: use noqa per line
    "E731",  # Do not assign a `lambda` expression, use a `def`
]
unfixable = ["F401"]

[tool.ruff.lint.pydocstyle]
# Use Google-style docstrings.
convention = "google"

#[tool.ruff.pycodestyle]
#ignore-overlong-task-comments = true

[tool.ruff.lint.mccabe]
# Unlike Flake8, default to a complexity level of 10.
max-complexity = 10

[tool.ruff.lint.pycodestyle]
max-line-length = 120

[dependency-groups]
# Developer-only dependency groups (not accessible via pip)
# For uv users: uv sync --group dev-all (includes all extras)
# For CI: uv sync --group dev-all --group torch-cpu (uses CPU-only PyTorch)
dev = [
    "pre-commit>=4.5.1",
    "pytest-cov>=7.0.0",
    "pytest-timeout>=2.3.0",
    "testfixtures>=8.3.0",
    "ty>=0.0.17",
]
torch-cpu = ["torch>=2.0.0"]
optional = [
    "onnx>=1.20.1",
    "onnxruntime>=1.24.1; python_version >= '3.11'",
    "onnxruntime<1.24.1; python_version < '3.11'",  # 1.24.1 dropped python 3.10 support
    "optuna>=4.7.0",
    "optuna-integration[pytorch-lightning]>=4.7.0",
    "polars>=1.37.1",
    "pydantic>=2.12.5",
    "ray>=2.53.0",
    "plotly>=6.5.2",
    "neuralforecast>=3.0.0",
]
release = [
    "bump-my-version>=1.2.5",
    "docutils==0.21.2",
    "ipykernel==7.1.0",
    "ipython==9.7.0; python_version >= '3.11'",
    "ipython==8.38.0; python_version < '3.11'",
    "ipython-genutils==0.2.0",
    "ipywidgets==8.1.8",
    "jinja2==3.1.6",
    "jupyterlab==4.4.10",
    "lxml-html-clean==0.4.3",
    "m2r2==0.3.4",
    "nbsphinx==0.9.7",
    "numpydoc==1.9.0",
    "papermill==2.6.0",
    "pydata-sphinx-theme==0.16.1",
    "recommonmark==0.7.1",
    "sphinx==8.1.3",
    "sphinx-autodoc-typehints==3.0.1",
    "sphinx-copybutton==0.5.2",
    "sphinx-design==0.6.1",
    "tenacity==9.1.2",
    "twine==6.2.0",
]
dev-all = [
    {include-group = "dev"},
    {include-group = "optional"},
    {include-group = "release"},
    "darts[all]",
]

[tool.bumpversion]
current_version = "0.43.0"
parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
serialize = ["{major}.{minor}.{patch}"]
commit = true
tag = true
tag_name = "{new_version}"
message = "Bump version: {current_version} → {new_version}"

[[tool.bumpversion.files]]
filename = "pyproject.toml"
search = 'version = "{current_version}"'
replace = 'version = "{new_version}"'

[[tool.bumpversion.files]]
filename = "darts/__init__.py"
search = '__version__ = "{current_version}"'
replace = '__version__ = "{new_version}"'

[[tool.bumpversion.files]]
filename = "docs/source/conf.py"
search = 'version = "{current_version}"'
replace = 'version = "{new_version}"'

[tool.ty.src]
include = ["darts"]
exclude = ["darts/tests"]

[tool.ty.terminal]
output-format = "concise"