gymnasium==0.28.1
ray[rllib]==2.31.0
protobuf==3.20.3
wandb
hydra-core
omegaconf
grpcio>=1.48.1
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
mlflow<3.0.0
yamale>=6.1.0

[:sys_platform != "darwin" or platform_machine != "arm64"]
mlagents-envs==1.1.0
gym==0.21.0
pettingzoo==1.15.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

[rl]
mlagents-envs==1.1.0
gym==0.21.0
pettingzoo==1.15.0
