pytorch-lightning<1.6,>=1.5.8
hydra-core
wandb
python-dotenv
stqdm
rich

[dev]
black
flake8
isort
pre-commit
bandit
pytest
pytest-cov

[test]
pytest
pytest-cov
