fix scikit-learn

swe-bench · Apr 1, 2024 · 35ab948 · 35ab948
1 parent 302f81f
commit 35ab948
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/harness/constants.py b/harness/constants.py
@@ -1,26 +1,26 @@
 MAP_VERSION_TO_INSTALL_SKLEARN = {
     k: {
         "python": "3.6",
-        "packages": "numpy scipy cython pytest pandas matplotlib",
-        "install": "python setup.py develop",
+        "packages": "numpy scipy Cython==0.27 pytest pandas matplotlib",
+        "install": "pip install -v --no-use-pep517 --no-build-isolation -e .",
     }
-    for k in ["0.20", "0.21", "0.22"]
+    for k in ["0.20"]
 }
 MAP_VERSION_TO_INSTALL_SKLEARN.update(
     {
         k: {
             "python": "3.7",
-            "packages": "numpy scipy cython pytest pandas matplotlib",
-            "install": "python setup.py develop",
+            "packages": "numpy scipy Cython==0.29 pytest pandas matplotlib",
+            "install": "pip install -v --no-use-pep517 --no-build-isolation -e .",
         }
-        for k in ["0.23", "0.24"]
+        for k in ["0.21", "0.22", "0.23", "0.24"]
     }
 )
 MAP_VERSION_TO_INSTALL_SKLEARN.update(
     {
         k: {
             "python": "3.9",
-            "packages": "numpy scipy cython pytest pandas matplotlib joblib threadpoolctl",
+            "packages": "numpy scipy Cython==0.29 pytest pandas matplotlib joblib threadpoolctl",
             "install": "pip install -v --no-use-pep517 --no-build-isolation -e .",
         }
         for k in ["1.0", "1.1", "1.2", "1.3", "1.4"]