Fix: Use swe-bench workaround for yanked pkgs

princeton-nlp · Oct 23, 2024 · a8ee3c5 · a8ee3c5
1 parent df210c0
commit a8ee3c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sweagent/environment/swe_env.py b/sweagent/environment/swe_env.py
@@ -22,6 +22,7 @@
 from simple_parsing.helpers.serialization.serializable import FrozenSerializable
 from swebench.harness.constants import MAP_REPO_VERSION_TO_SPECS
 from swebench.harness.utils import get_environment_yml, get_requirements
+from swebench.swebench.harness.test_spec import replace_uninstallable_packages_requirements_txt
 
 import docker
 import docker.errors
@@ -1224,7 +1225,7 @@ def install_env(self) -> None:
                 )
                 self.logger.debug("Created conda environment")
                 # Write reqs to requirements.txt in docker container
-                content_reqs = get_requirements(self.record)
+                content_reqs = replace_uninstallable_packages_requirements_txt(get_requirements(self.record))
                 copy_file_to_container(self.container_obj, content_reqs, PATH_TO_REQS)
                 # Create conda environment + install reqs
                 self.communicate_with_handling(