numpy>=2.0.0
scikit-learn
k-means-constrained
torch>=2.0.0
tqdm

[dev]
pytest
black
isort
flake8
