update setup.py

csinva · Jan 5, 2021 · 0d3bc1a · 0d3bc1a
1 parent a1aa4ac
commit 0d3bc1a
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",