From df72c9330631f643b8fe644f25f4a822d16cd4ac Mon Sep 17 00:00:00 2001
From: Kade Heckel <51248168+kmheckel@users.noreply.github.com>
Date: Mon, 29 Jul 2024 13:00:06 +0100
Subject: [PATCH] Update setup.py

fixed deprecated pip install sklearn and made torchvision the cpu edition.
---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 3e7f7dd..b638a70 100644
--- a/setup.py
+++ b/setup.py
@@ -24,8 +24,8 @@
 extras = {
     'loaders' : [
         'tonic',
-        'torchvision',
-        'sklearn'
+        'torchvision==0.19.0+cpu',
+        'scikit-learn'
     ]
 }