torch>=2.1.0
torchvision>=0.16.0
click>=8.0.0
einops>=0.8.2
tqdm>=4.67.1
pyyaml>=6.0.3

[all]
simulusion[brax,bsuite,dev,dmc,docs,gym,ml,ml-agents,mujoco,robotics,viz,worldmodels]

[brax]
brax>=0.13.0
jax>=0.10.1

[bsuite]
bsuite>=0.3.5

[dev]
pytest>=9.0.3
pytest-cov>=7.1.0
pytest-timeout>=2.3.1
pytest-xdist>=3.6.0
pre-commit>=4.5.0
mypy>=1.19.1
gitpython>=3.1.58
ruff==0.16.8

[dmc]
gymnasium>=1.2.2

[dmc-uv]
dm-control>=1.0.0
gymnasium>=1.2.2

[dmc-uv:python_version >= "3.13"]
labmaze-new>=0.0.2

[dmc:python_version < "3.13"]
dm-control>=1.0.0

[dmc:python_version >= "3.13"]
labmaze-new>=0.0.2

[docs]
sphinx<9.0.0,>=8.0.0
sphinx-autodoc-typehints<1.26.0
sphinx-copybutton>=0.5.2
pydata-sphinx-theme>=0.15.0
sphinx-rtd-theme>=3.1.0
sphinxcontrib-bibtex>=2.6.5
sphinxcontrib-mermaid>=0.9.0
sphinx-thebe>=0.3.0
sphinxext-opengraph>=0.13.0
myst-parser<5.0.0,>=2.0.0
nbsphinx>=0.9.8
mistune>=3.3.0
nbconvert>=7.17.1

[genie]
h5py>=3.11.0
huggingface-hub>=0.23.0
omegaconf>=2.3.0

[gym]
gymnasium[box2d]>=1.2.2
huggingface-hub>=0.23.0
ale-py>=0.10.1
autorom>=0.6.1

[ml]
torchaudio>=0.13.0
hydra-core>=1.3.2
omegaconf>=2.3.0
wandb>=0.16.0
tensorboard>=2.20.0
nvidia-ml-py>=12.535.77
tensorboardx>=2.6.4

[ml-agents]
mlagents-envs<0.29.0,>=0.28.0
protobuf>=5.29.6

[mujoco]
mujoco>=3.2.0

[robotics]
gymnasium-robotics>=1.3.0
mujoco<3.12,>=3.2.0

[viz]
opencv-python>=4.12.0.88
imageio-ffmpeg>=0.5.1
umap-learn>=0.5.11
scikit-learn>=1.8.0
plotly>=6.5.0
h5py>=3.11.0

[worldmodels]
albumentations>=2.0.0
cma>=4.0.0
opencv-python>=4.12.0.88
imageio-ffmpeg>=0.5.1
