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

[dev]
pytest
pre-commit
black
ruff

[train]
pandas
scikit-learn
