Merge pull request #226 from wistuba/patch-1

fix docker building issues with sklearn
swe-bench · Oct 7, 2024 · ba221bb · ba221bb
2 parents 0f99aef + 9738a50
commit ba221bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/swebench/harness/constants.py b/swebench/harness/constants.py
@@ -76,7 +76,7 @@ class TestStatus(Enum):
     {
         k: {
             "python": "3.9",
-            "packages": "numpy scipy cython setuptools pytest pandas matplotlib joblib threadpoolctl",
+            "packages": "'numpy==1.19.2' 'scipy==1.5.2' 'cython==3.0.10' pytest 'pandas<2.0.0' 'matplotlib<3.9.0' setuptools pytest joblib threadpoolctl",
             "install": "python -m pip install -v --no-use-pep517 --no-build-isolation -e .",
             "pip_packages": ["cython", "setuptools", "numpy", "scipy"],
             "test_cmd": TEST_PYTEST,