torch>=2.0
numpy>=1.21

[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

[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

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

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

[umap]
umap-learn>=0.5
