torch
scikit-learn
tqdm

[dev]
pytest>=7.0
