torch<3.0.0,>=2.0.0
numpy<3.0.0,>=1.24.0

[all]
worldflux[atari,dev,docs,env,logging,training,viz]

[atari]
gymnasium[atari]<2.0.0,>=0.29.0
ale-py<1.0.0,>=0.8.0
opencv-python<5.0.0,>=4.0.0

[dev]
pytest<9.0.0,>=7.0.0
pytest-cov<6.0.0,>=4.0.0
mypy<2.0.0,>=1.0.0
ruff<1.0.0,>=0.1.0
pre-commit<5.0.0,>=3.0.0

[docs]
mkdocs<2.0.0,>=1.5.0
mkdocs-material<10.0.0,>=9.0.0
mkdocstrings[python]<1.0.0,>=0.24.0
pymdown-extensions<11.0.0,>=10.0.0

[env]
gymnasium<2.0.0,>=0.29.0

[logging]
wandb<1.0.0,>=0.15.0

[mujoco]
gymnasium[mujoco]<2.0.0,>=0.29.0
mujoco<4.0.0,>=3.0.0

[training]
tqdm<5.0.0,>=4.0.0

[viz]
matplotlib<4.0.0,>=3.5.0
scikit-learn<2.0.0,>=1.0.0
