pyproject.toml

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

[tool.hatch.version]
path = "src/riemax/__init__.py"

[tool.hatch.build.targets.sdist]
exclude = [
  "/docs",
]

[tool.hatch.build.targets.wheel]
packages = ["src/riemax"]

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

[project]
name = "riemax"
description = "Riemannian geometry in JAX."

requires-python = ">=3.12"
readme = "README.md"
license = { file = "LICENSE" }

authors = [
    { name = "Daniel Kelshaw", email = "daniel.j.kelshaw@gmail.com" },
]

keywords = ["jax", "geometry", "deep-learning"]

dynamic = ["version"]

dependencies = [
    "absl-py",
    "chex",
    "dm-haiku",
    "einops",
    "jax",
    "jaxlib",
    "matplotlib",
    "numpy",
    "scipy",
    "optax",
    "ml_collections @ git+https://github.com/danielkelshaw/ml_collections",
    "opt_einsum @git+https://github.com/dgasmith/opt_einsum",
]

[project.optional-dependencies]

jupyter = [
    "ipykernel",
    "jupyterlab",
]

dev = [
    "mkdocs",
    "mkdocs-material",
    "mkdocstrings-python",
    "mknotebooks",
    "pre-commit",
    "pyright == 1.1.332",
    "ruff",
]

[project.urls]
homepage = "https://riemax.danielkelshaw.com"
repository = "https://github.com/danielkelshaw/riemax"

[tool.ruff]
indent-width = 4
line-length = 120
target-version = "py312"

[tool.ruff.lint]
ignore = ["E501", "B023"]
select = ["E", "F", "B", "I"]

[tool.ruff.lint.per-file-ignores]
"**/__init__.py" = ["F401"]

[tool.ruff.format]
quote-style = "single"
indent-style = "space"
docstring-code-format = true

[tool.pyright]
include = ["src"]
ignore = ["src/riemax/ml/eikonal/model.py"]
pythonVersion = "3.12"
venvPath = "."
venv = ".venv"

useLibraryCodeForTypes = true
reportMissingImports = true