torch>=1.9.0
tqdm>=4.0.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
scikit-learn>=1.0.0
numpy>=1.20.0
