From 7b23983887339a8bb96fca180a982f94e30b9c15 Mon Sep 17 00:00:00 2001
From: rmschubert <ronny.schubert59@gmail.com>
Date: Mon, 5 Dec 2022 16:31:33 +0100
Subject: [PATCH] fix: update scikit-learn dependency

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

diff --git a/setup.py b/setup.py
index 8b22cfa..54b508d 100644
--- a/setup.py
+++ b/setup.py
@@ -22,7 +22,7 @@ INSTALL_REQUIRES = [
     "torch>=1.3.1",
     "torchvision>=0.7.4",
     "numpy>=1.9.1",
-    "sklearn",
+    "scikit-learn",
     "matplotlib",
 ]
 DATASETS = [