fix astropy for v5+

swe-bench · Apr 1, 2024 · 302f81f · 302f81f
1 parent 9b8dde6
commit 302f81f
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/harness/constants.py b/harness/constants.py
@@ -216,15 +216,24 @@
 
 MAP_VERSION_TO_INSTALL_ASTROPY = {
     k: {
-        "python": "3.9", 
+        "python": "3.9",
         "install": "python setup.py install",
         "pip_packages": "pytest extension-helpers numpy",
-        }
-    for k in
-        ["0.1", "0.2", "0.3", "0.4", "1.1", "1.2", "1.3", "3.0", "3.1", "3.2"] + \
-        ["4.1", "4.2", "4.3", "5.0", "5.1", "5.2"]
+    }
+    for k in ["0.1", "0.2", "0.3", "0.4", "1.1", "1.2", "1.3", "3.0", "3.1", "3.2"]
 }
 
+MAP_VERSION_TO_INSTALL_ASTROPY.update(
+    {
+        k: {
+            "python": "3.9",
+            "install": "pip install -e .[test]",
+            "pip_packages": "pytest extension-helpers numpy",
+        }
+        for k in ["4.1", "4.2", "4.3", "5.0", "5.1", "5.2"]
+    }
+)
+
 MAP_VERSION_TO_INSTALL_SYMPY = {
     k: {
         "python": "3.9",