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 +)