numpy>=1.20.0
pyyaml>=5.1
torch>=1.9.0
matplotlib>=3.3.0
optuna>=3.0.0

[all]
rllama[dev,gym,rlhf,sb3,vis]

[dev]
pytest>=6.0.0
black>=22.3.0
isort>=5.10.0
build
twine

[gym]
gym>=0.17.0

[rlhf]
tqdm>=4.45.0

[sb3]
stable-baselines3>=1.5.0

[vis]
streamlit>=1.10.0
