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.29.*
gymnasium-minigrid
gymnasium-robotics

[faiss]
faiss-cpu>=1.7.4

[faiss_gpu]
faiss-gpu>=1.7.4
