pyproject.toml

[build-system]
requires = ["setuptools>61", "wheel", "toml", "build"]
build-backend = "setuptools.build_meta"

[project]
name = "aeon"
version = "0.11.1"
description = "A toolkit for machine learning from time series"
authors = [
    {name = "aeon developers", email = "contact@aeon-toolkit.org"},
]
maintainers = [
    {name = "aeon developers", email = "contact@aeon-toolkit.org"},
]
readme = "README.md"
keywords = [
    "data-science",
    "machine-learning",
    "data-mining",
    "time-series",
    "scikit-learn",
    "forecasting",
    "time-series-analysis",
    "time-series-classification",
    "time-series-clustering",
    "time-series-regression",
]
classifiers = [
    "Intended Audience :: Science/Research",
    "Intended Audience :: Developers",
    "License :: OSI Approved :: BSD License",
    "Programming Language :: Python",
    "Topic :: Software Development",
    "Topic :: Scientific/Engineering",
    "Development Status :: 5 - Production/Stable",
    "Operating System :: Microsoft :: Windows",
    "Operating System :: POSIX",
    "Operating System :: Unix",
    "Operating System :: MacOS",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
]
requires-python = ">=3.9,<3.13"
dependencies = [
    "deprecated>=1.2.13",
    "numba>=0.55,<0.61.0",
    "numpy>=1.21.0,<1.27.0",
    "packaging>=20.0",
    "pandas>=1.5.3,<2.1.0",
    "scikit-learn>=1.0.0,<1.6.0",
    "scipy>=1.9.0,<1.13.0",
    "typing-extensions>=4.6.0",
]

[project.optional-dependencies]
all_extras = [
    "cloudpickle",
    "esig>=0.9.7; platform_system != 'Darwin' and python_version < '3.11'",
    "gluonts>=0.12.4",
    "h5py",
    "imbalanced-learn",
    "keras-self-attention",
    "kotsu>=0.3.1",
    "matplotlib>=3.3.2,<3.9.0",
    "pmdarima>=1.8.0",
    "prophet>=1.1.0",
    "pydot>=2.0.0",
    "scikit_posthocs>=0.6.5",
    "seaborn>=0.11.0",
    "statsforecast>=0.5.2",
    "statsmodels>=0.12.1",
    "stumpy>=1.5.1",
    "ruptures>=1.1.9",
    "tbats>=1.1.0",
    "tensorflow>=2.14; python_version < '3.12'",
    "torch>=1.13.1",
    "tsfresh>=0.20.0",
    "tslearn>=0.5.2",
    "xarray",
    "pyod>=1.1.3",
    "prts>=1.0.0.0",

    # for some reason, a heirarchical dask test fails on some Python versions with the
    # later versions of pyarrow and dask
    "dask[dataframe]<2024.3.0; python_version < '3.11'",
    "pyarrow<12; python_version < '3.12'",
    "pyarrow-hotfix; python_version < '3.12'",
]
dl = [
    "keras-self-attention",
    "tensorflow>=2.14; python_version < '3.12'",
]
unstable_extras = [
    "mrsqm>=0.0.7,<0.1.0; platform_system != 'Windows' and python_version < '3.12'",  # requires gcc and fftw to be installed for Windows and some other OS (see http://www.fftw.org/index.html)
    "pycatch22",  # known to fail installation on some setups
    "pyfftw>=0.12.0; python_version < '3.12'",  # requires fftw to be installed for Windows and some other OS (see http://www.fftw.org/index.html)
]
dev = [
    "backoff",
    "httpx",
    "pre-commit",
    "pytest",
    "pytest-cov",
    "pytest-mock",
    "pytest-randomly",
    "pytest-timeout",
    "pytest-xdist[psutil]",
    "pytest-rerunfailures",
    "wheel",
]
binder = [
    "notebook",
    "jupyterlab",
]
docs = [
    "sphinx<7.4.0",
    "sphinx-design",
    "sphinx-version-warning",
    "sphinx_issues",
    "sphinx-copybutton",
    "sphinx-remove-toctrees",
    "sphinxext-opengraph",
    "nbsphinx",
    "numpydoc",
    "myst-parser",
    "jupyter",
    "furo",
    "tabulate",
    "matplotlib",
]

[project.urls]
Homepage = "https://www.aeon-toolkit.org"
Repository = "https://github.com/aeon-toolkit/aeon"
Documentation = "https://www.aeon-toolkit.org"
Changelog = "https://www.aeon-toolkit.org/en/stable/changelog.html"
Download = "https://pypi.org/project/aeon/#files"
API = "https://www.aeon-toolkit.org/en/stable/api_reference.html"

[project.license]
file = "LICENSE"

[tool.setuptools]
zip-safe = true

[tool.setuptools.package-data]
aeon = [
    "*.csv",
    "*.csv.gz",
    "*.arff",
    "*.arff.gz",
    "*.txt",
    "*.ts",
    "*.tsv",
]

[tool.setuptools.packages.find]
exclude = ["tests", "tests.*"]

[tool.check-manifest]
ignore = [
    # Ignore virtual environments in local builds
    "venv/**",
    # Ignore the docs symlink and its contents
    "docs/examples",
    "docs/examples/**",
]

[tool.ruff.lint]
select = ["D"]

[tool.ruff.lint.pydocstyle]
convention = "numpy"

[tool.pytest.ini_options]
# ignore certain folders and pytest warnings
testpaths = "aeon"
addopts = '''
    --doctest-modules
    --durations 20
    --timeout 600
    --showlocals
    --dist worksteal
    --reruns 2
    --only-rerun "crashed while running"
'''
filterwarnings = '''
    ignore::UserWarning
    ignore:numpy.dtype size changed
    ignore:numpy.ufunc size changed
'''