numpy
tensorboard
wget
huggingface_hub>=0.10
gymnasium
stable-baselines3
huggingface_sb3
onnx
onnxruntime

[all]
numpy==1.23.5
sample-factory
ray==2.2.0
ray[rllib]
tensorflow_probability

[clean-rl]
wandb

[dev]
pytest>=6.0
pytest-xdist
black[jupyter]~=22.0
flake8>=3.8.3
isort>=5.0.0
pyyaml>=5.3.1

[rllib]
numpy==1.23.5
ray==2.2.0
ray[rllib]
tensorflow_probability

[sf]
sample-factory

[test]
pytest>=6.0
pytest-xdist
