Merge pull request #26 from csinva/update_requirements

update setup.py
csinva · Jan 5, 2021 · 4dd5634 · 4dd5634
2 parents a1aa4ac + 0d3bc1a
commit 4dd5634
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -15,13 +15,14 @@
         'scipy',
         'matplotlib',
         'pandas',
-        'scikit-learn',
+        'scikit-learn>=0.23.0', # 0.23+ only works on py3.6+)
         'cvxpy',
         'cvxopt',
         'mlxtend',
         'pytest',
         'pytest-cov'
     ],
+    python_requires='>=3.6',
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: MIT License",