Update setup.py

fixed deprecated pip install sklearn and made torchvision the cpu edition.
kmheckel · Jul 29, 2024 · df72c93 · df72c93
1 parent 13eccaf
commit df72c93
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -24,8 +24,8 @@
 extras = {
     'loaders' : [
         'tonic',
-        'torchvision',
-        'sklearn'
+        'torchvision==0.19.0+cpu',
+        'scikit-learn'
     ]
 }