mlagents==1.1.0
mlagents-envs==1.1.0
gymnasium==0.28.1
gym==0.21.0
ray[rllib]==2.31.0
protobuf==3.20.3
wandb
hydra-core
omegaconf
grpcio<=1.48.2,>=1.11.0
numpy<1.24.0,>=1.23.5
torch>=2.1.1
pybind11>=2.12.0
attrs>=19.3.0
cattrs<1.7,>=1.1.0
h5py>=2.9.0
huggingface-hub>=0.14
onnx==1.15.0
Pillow>=4.2.1
pyyaml>=3.1.0
tensorboard>=2.14
pettingzoo==1.15.0
mlflow<3.0.0
yamale>=6.1.0

[all]
replicantdrivesim[dev,docs,mlflow]

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
isort>=5.0.0
flake8>=6.0.0
mypy>=1.0.0

[docs]
sphinx>=7.0.0
sphinx-rtd-theme>=1.3.0
myst-parser>=2.0.0

[mlflow]
mlflow>=1.27.0
tensorboardx>=2.6.0
