fix: update scikit-learn dependency

This commit is contained in:
rmschubert 2022-12-05 16:31:33 +01:00 committed by Alexander Engelsberger
parent 0649d5bb45
commit 7b23983887

View File

@ -22,7 +22,7 @@ INSTALL_REQUIRES = [
"torch>=1.3.1",
"torchvision>=0.7.4",
"numpy>=1.9.1",
"sklearn",
"scikit-learn",
"matplotlib",
]
DATASETS = [