torch>=2.0
numpy>=1.21

[atari]
gymnasium[atari]==0.28.1
autorom>=0.6

[dev]
pytest>=7.0
pytest-cov>=4.0
black>=23.7
ruff>=0.1.7
isort>=5.12
mypy>=1.7
pre-commit>=3.5
hypothesis>=6.80
sympy>=1.14
scipy>=1.14

[docs]
mkdocs>=1.5
mkdocstrings[python]>=0.24
mkdocs-material>=9.5

[envs]
gymnasium==0.28.1
minigrid>=3.0.0
pygame>=2.4.0

[faiss]
faiss-cpu>=1.7.4

[faiss_gpu]
faiss-gpu>=1.7.4

[gym]
gymnasium==0.28.1

[hub]
huggingface_hub>=0.20
safetensors>=0.4
jsonschema>=4.0

[math]
sympy>=1.14
scipy>=1.14
hypothesis>=6.80

[min_torch]
numpy<2
torch==2.0.*

[release]
build>=1.0
twine>=4.0

[robotics]
gymnasium-robotics>=1.4.2
mujoco>=2.3.0

[test]
pytest>=7.0
pytest-cov>=4.0
hypothesis>=6.80

[umap]
umap-learn>=0.5

[viz]
matplotlib>=3.7
