torch>=2.0
numpy
click>=8.0
tqdm>=4.0

[dev]
pytest
pre-commit
black
ruff
cogapp

[train]
pandas
scikit-learn
lightning
