pyproject.toml

[project]
name = "exponax"
version = "0.2.0"
description = "Efficient differentiable PDE solvers in JAX."
readme = "README.md"
requires-python ="~=3.10"
authors = [
  {name = "Felix Koehler"},
]
keywords = ["jax", "sciml", "deep-learning", "pde", "etdrk"]
urls = {repository = "https://github.com/Ceyron/exponax" }
dependencies = ["jax>=0.4.13", "jaxtyping>=0.2.20", "typing_extensions>=4.5.0", "equinox>=0.11.3", "matplotlib>=3.8.1"]

[project.optional-dependencies]
vape4d = ["vape4d"]
test = ["jaxlib", "pytest", "pytest-cov"]
docs = [
    "mkdocs==1.6.1",
    "mkdocs-material==9.7.1",
    "mkdocstrings==1.0.3",
    "mkdocstrings-python==2.0.1",
    "mknotebooks==0.8.0",
    "griffe==1.15.0",
    "black==26.1.0",  # Needed for API autoformat
]

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

[tool.ruff.lint]
select = [
    "E",   # pycodestyle errors
    "W",   # pycodestyle warnings
    "F",   # pyflakes
    "I",   # isort
    "UP",  # pyupgrade
    "B",   # flake8-bugbear
    "SIM", # flake8-simplify
]
ignore = [
    "F722",  # jaxtyping uses non-standard annotation syntax
]

[tool.ruff.lint.isort]
known-first-party = ["exponax"]

[tool.setuptools.packages.find]
where = ["."]  # list of folders that contain the packages (["."] by default)
include = ["exponax*"]  # package names should match these glob patterns (["*"] by default)
exclude = []  # exclude packages matching these glob patterns (empty by default)
namespaces = false  # to disable scanning PEP 420 namespaces (true by default)

[tool.coverage.run]
source = ["exponax"]

[tool.coverage.report]
exclude_lines = [
    "pragma: no cover",
    "if __name__",
    "raise NotImplementedError",
    "@abstractmethod",
]