torch>=1.7.0
numpy>=1.19.0
scikit-learn>=0.24.0
imbalanced-learn>=0.8.0

[dev]
pytest
pytest-cov
flake8
black
isort

[docs]
sphinx
sphinx-rtd-theme
