pyproject.toml

[project]
name = "jaximal"
version = "0.1.1"
description = "A JAX-based PyTree manipulation library "
authors = [
  { name = "Arvin Kushwaha", email = "arvin.singh.kushwaha@gmail.com" },
]
dependencies = [
  "safetensors>=0.4.3",
  "jaxtyping>=0.2.29",
  "jax>=0.4.28",
  "jaxlib>=0.4.28",
]
readme = "README.md"
requires-python = ">= 3.12"
license = { text = "MIT" }

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

[tool.rye]
managed = true
dev-dependencies = ["pytest>=8.2.1", "basedpyright>=1.12.4", "optax>=0.2.2"]
excluded-dependencies = []

[tool.rye.scripts]
ci = { chain = [
  # "ci:verifytypes", 
  "ci:basedpyright",
] }
# "ci:verifytypes" = "rye run basedpyright --verifytypes jaximal"
"ci:basedpyright" = "rye run basedpyright -p . ."

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

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

[tool.ruff]
target-version = "py312"

[tool.ruff.lint.isort]
lines-between-types = 1
relative-imports-order = "closest-to-furthest"

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

[tool.basedpyright]
typeCheckingMode = "strict"
reportMissingTypeStubs = false
reportUnknownParameterType = false
reportUnknownArgumentType = false
reportUnknownMemberType = false
reportUnknownVariableType = false