pyproject.toml

[project]
name = "dynax"
version = "0.0.3"
description = "Dynamical systems with JAX!"
readme = "README.rst"
requires-python = ">=3.10"
license = { file = "LICENSE" }
authors = [{ name = "Franz M. Heuchel", email = "franz.heuchel@pm.me" }]
keywords = [
    "jax",
    "dynamical-systems",
    "system-identification",
    "linearization",
]
urls = { repository = "https://github.com/fhchl/dynax" }
dependencies = ["jax<=0.4.33", "diffrax<=0.6"]

[project.optional-dependencies]
dev = ["pytest", "jupyter", "matplotlib", "pre-commit", "ruff<0.6"]
docs = [
    "nbsphinx",
    "sphinx-autobuild",
    "sphinx-autodoc-typehints",
    "sphinx-rtd-theme",
    "sphinx",
    "sphinxcontrib-bibtex",
    "sphinxcontrib-aafig",
    "furo",
]

[tool.pytest.ini_options]
addopts = [
    "--pdbcls=IPython.terminal.debugger:Pdb",
    # "--jaxtyping-packages=dynax,beartype.beartype(conf=beartype.BeartypeConf(strategy=beartype.BeartypeStrategy.On))"
]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.ruff]
line-length = 88
src = ["dynax", "tests", "examples"]
force-exclude = true
extend-include = ["*.ipynb"]

[tool.ruff.lint]
select = ["E", "F", "I001", "B"]
ignore = [
    "E402", # Module level import not at top of file
    "E721", # Do not compare types, use 'isinstance()'
    "E731", # Do not assign a lambda expression, use a def (E731)
    "E741", # Do not use variables named 'I', 'O', or 'l'
]
fixable = ["I001", "F401"]

[tool.ruff.lint.isort]
combine-as-imports = true
lines-after-imports = 2
order-by-type = false
known-first-party = ["dynax"]

[tool.uv]
dev-dependencies = [
    "dynax[dev, docs]",
]

[tool.uv.sources]
dynax = { workspace = true }