From 174909e0c541b8e7b3a2e86c79329ea27184e463 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Steffen=20R=C3=B6cker?= <sroecker@gmail.com>
Date: Thu, 7 Sep 2023 15:06:57 +0200
Subject: [PATCH] Require scikit-learn instead of sklearn package

Fixes #48
---
 requirements.txt | 3 +--
 setup.py         | 3 +--
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index cfb8e6d..d7c8928 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,9 +1,8 @@
 ray~=1.3.0
 setuptools~=52.0.0
-sklearn~=0.0
 scikit-learn~=0.24.1
 pandas~=1.2.4
 pytest~=6.2.4
 numpy~=1.18.5
 pickle5~=0.0.11
-graphviz~=0.16
\ No newline at end of file
+graphviz~=0.16
diff --git a/setup.py b/setup.py
index 58b6920..d06b5c5 100644
--- a/setup.py
+++ b/setup.py
@@ -41,7 +41,6 @@
     install_requires=[
         'ray[default,serve,k8s]>=1.3.0',
         'setuptools>=52.0.0',
-        'sklearn>=0.0',
         'scikit-learn>=0.24.1',
         'pandas>=1.2.4',
         'numpy>=1.18.5',
@@ -69,4 +68,4 @@
         'Bug Reports': 'https://github.com/project-codeflare/codeflare/issues',
         'Source': 'https://github.com/project-codeflare/codeflare',
     },
-)
\ No newline at end of file
+)