click>=8.0.0
pyyaml>=6.0
wandb>=0.12.0
prompt_toolkit>=3.0.0
typing-extensions>=4.0.0

[dev]
bump2version>=1.0.1
twine>=4.0.0
build>=1.0.0

[docs]
mkdocs>=1.4.0
mkdocs-material>=9.0.0
mkdocstrings[python]>=0.20.0

[test]
pytest>=7.0.0
pytest-cov>=4.0.0
pytest-mock>=3.0.0
torch
