From 45a3339e289def93ead55a1c83de72934147fb4a Mon Sep 17 00:00:00 2001
From: ahalev <avishaihalev@gmail.edu>
Date: Tue, 30 Apr 2024 15:22:57 -0700
Subject: [PATCH] update sklearn dependency

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index c0689e3..058b8ca 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@
         "opencv-python",
         "torchvision>=0.4.2",
         "Pillow",
-        "sklearn",
+        "scikit-learn",
         "fairscale",
     ],
     packages=find_packages(exclude=("configs", "tests")),