pyproject.toml

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

[project]
name = "go1-ppo"
version = "0.1.0"
description = "PPO locomotion training for the Unitree Go1 quadruped with JAX, Brax, and MuJoCo MJX"
readme = "README.md"
requires-python = ">=3.12,<3.13"
license = { text = "Apache-2.0 AND BSD-3-Clause" }
license-files = ["LICENSE", "THIRD_PARTY_NOTICES.md"]
keywords = [
    "unitree-go1",
    "go1",
    "quadruped",
    "legged-locomotion",
    "reinforcement-learning",
    "ppo",
    "mujoco",
    "mjx",
    "jax",
    "brax",
    "robotics",
    "domain-randomization",
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: Apache Software License",
    "License :: OSI Approved :: BSD License",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
    "absl-py",
    "brax>=0.12.5",
    "etils",
    "flax",
    "jax[cuda12]",
    "lxml",
    "mediapy",
    "ml_collections",
    "mujoco-mjx>=3.4.dev",
    "mujoco>=3.4.dev",
    "orbax-checkpoint>=0.11.22",
    "setuptools>=80,<81",
    "tensorboard>=2.20.0",
    "tensorboardX",
    "tqdm",
    "warp-lang>=1.12.0",
]

[[tool.uv.index]]
name = "mujoco"
url = "https://py.mujoco.org"
explicit = true

[[tool.uv.index]]
name = "nvidia"
url = "https://pypi.nvidia.com"
explicit = true

[tool.uv.sources]
mujoco = { index = "mujoco" }
mujoco-mjx = { git = "https://github.com/google-deepmind/mujoco", rev = "main", subdirectory = "mjx" }
warp-lang = { index = "nvidia" }

[project.optional-dependencies]
dev = [
    "pytest",
    "ruff",
]
learning = [
    "wandb",
]

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

[tool.hatch.build.targets.wheel]
packages = ["go1_ppo"]

[tool.hatch.build]
include = [
    "go1_ppo/**/*",
    "LICENSE",
    "THIRD_PARTY_NOTICES.md",
]