84 lines
1.5 KiB
TOML
84 lines
1.5 KiB
TOML
[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
|