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

[project]
name = "2026sel3-project"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">= 3.12, < 3.13"
dependencies = [
    "biorobot==0.4.2",
    "cleanrl>=0.4.8",
    "evosax==0.2.0",
    "flax>=0.12.2",
    "gymnasium>=1.2.3",
    "ipykernel==7.2.0",
    "jax==0.9.0.1",
    "numpy>=2.0.0",
    "protobuf>=5.0.0",
    "warp-lang",
    "mujoco-warp",
    "matplotlib==3.10.8",
    "mediapy==1.2.6",
    "optax>=0.2.6",
    "pyopengl>=3.1.10",
    "pyopengl-accelerate>=3.1.10",
    "pyyaml>=6.0",
    "hydra-core>=1.3.2",
    "wandb==0.24.2",
    "torch>=2.4.0",
]

[project.optional-dependencies]
cuda = [
    "jax[cuda13]==0.9.0.1",
]
analysis = [
    "tensorboard",
]
evaluation = [
    "imageio>=2.35.0",
    "imageio-ffmpeg>=0.5.1",
]

[dependency-groups]
dev = [
    "pre-commit>=4.0.0",
    "pytest>=8.0.0",
    "ruff>=0.15.2",
]

[tool.hatch.build.targets.wheel]
packages = ["src/brittle_star_project", "src/experiment_logger"]

[tool.mypy]
mypy_path = "src"
check_untyped_defs = false
warn_return_any = false

[[tool.mypy.overrides]]
module = [
    "jax.*",
    "flax.*",
    "wandb.*",
    "torch.*",
    "mujoco.*",
    "mujoco_warp.*",
    "optax.*",
    "tyro.*",
    "biorobot.*",
    "gymnasium.*",
    "matplotlib.*",
    "mediapy.*",
    "matplotlib.*",
    "mediapy.*",
    "pytest.*",
    "tensorboard.*",
    "tqdm.*",
    "numpy.*",
    "yaml.*",
    "moojoco.*"
]
ignore_missing_imports = true