pyproject.toml

[build-system]
requires = ["hatchling", "hatch-vcs"]
build-backend = "hatchling.build"

[project]
name = "pymc-extras"
authors = [
  {name = "PyMC Developers", email = "pymc.devs@gmail.com"},
]
description =  "A home for new additions to PyMC, which may include unusual probability distribitions, advanced model fitting algorithms, or any code that may be inappropriate to include in the pymc repository, but may want to be made available to users."
classifiers = [
  "Development Status :: 5 - Production/Stable",
  "Programming Language :: Python",
  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: 3.13",
  "License :: OSI Approved :: Apache Software License",
  "Intended Audience :: Science/Research",
  "Topic :: Scientific/Engineering",
  "Topic :: Scientific/Engineering :: Mathematics",
  "Operating System :: OS Independent",
]
readme = "README.md"
requires-python = ">=3.11"
keywords = [
  "probability",
  "machine learning",
  "statistics",
  "mcmc",
  "sampling",
  "bayesian",
]
license = {file = "LICENSE"}
dynamic = ["version"]  # specify the version in the __init__.py file
dependencies = [
  "pymc>=5.26.1",
  "pytensor>=2.35.1",
  "scikit-learn",
  "better-optimize>=0.1.5",
  "pydantic>=2.0.0",
  "preliz>=0.20.0",
]

[project.optional-dependencies]
complete = [
  "dask[complete]<2025.1.1",
  "xhistogram",
]
dev = [
  "pytest>=6.0",
  "pytest-mock",
  "dask[all]<2025.1.1",
  "blackjax",
  "statsmodels",
]
docs = [
  "nbsphinx>=0.4.2",
  "sphinx>=4.0",
  "pymc-sphinx-theme>=0.16",
]
dask_histogram = [
  "dask[complete]<2025.1.1",
  "xhistogram",
]
histogram = [
  "xhistogram",
]

[project.urls]
Documentation = "https://pymc-extras.readthedocs.io/"
Repository = "https://github.com/pymc-devs/pymc-extras.git"
Issues = "https://github.com/pymc-devs/pymc-extras/issues"

[tool.hatch.version]
source = 'vcs'

[tool.hatch.build.hooks.vcs]
version-file = "_version.py"

[tool.hatch.build.targets.sdist]
exclude = [
  '/.github',
  '/notebooks',
]

[tool.hatch.build.targets.wheel]
packages = ["pymc_extras"]

[tool.hatch.metadata]
allow-direct-references = true

[tool.pytest.ini_options]
minversion = "6.0"
xfail_strict = true
addopts = [
  "-v",
  "--doctest-modules",
  "--ignore=pymc_extras/model_builder.py"
]

filterwarnings =[
    "error",
    # JAX issues an over-eager warning if os.fork() is called when the JAX module is loaded, even if JAX isn't being used
    'ignore:os\.fork\(\) was called\.:RuntimeWarning',

    # Preliz needs to update for pytensor > 2.35
   'ignore:.*`pytensor\.graph\.basic\.ancestors`.*`pytensor\.graph\.traversal\.ancestors`.*:FutureWarning:^preliz(\.|$)',

    # OpenMP library warning on windows CI
  'ignore::RuntimeWarning:threadpoolctl'
]

[tool.coverage.report]
exclude_lines = [
  "pragma: nocover",
  "raise NotImplementedError",
  "if TYPE_CHECKING:",
]

[tool.ruff]
line-length = 100
target-version = "py311"

[tool.ruff.format]
docstring-code-format = true

[tool.ruff.lint]
select = ["D", "E", "F", "I", "UP", "W", "RUF"]
ignore = [
  "E501",
  "RUF001", # String contains ambiguous character (such as Greek letters)
  "RUF002", # Docstring contains ambiguous character (such as Greek letters)
  "RUF012", # Mutable class attributes should be annotated with `typing.ClassVar`
  "D100",
  "D101",
  "D102",
  "D103",
  "D104",
  "D105",
  "D107",
  "D200",
  "D202",
  "D203",
  "D204",
  "D205",
  "D209",
  "D212",
  "D213",
  "D301",
  "D400",
  "D401",
  "D403",
  "D413",
  "D415",
  "D417",
]

[tool.ruff.lint.isort]
lines-between-types = 1

[tool.ruff.lint.per-file-ignores]
'tests/*.py' = [
  'F841', # Unused variable warning for test files -- common in pymc model declarations
  'D106',  # Missing docstring for public method -- unittest test subclasses don't need docstrings
  'E402'  # Import at top, not respected when pytest.importorskip is required
]
'tests/statespace/*.py' =  [
  'F401', # Unused import warning for test files -- this check removes imports of fixtures
  'F811'  # Redefine while unused -- this check fails on imported fixtures
]
'pymc_extras/__init__.py' = ['F401', 'F403']