fix bug when remove images

princeton-nlp · Aug 13, 2024 · 00d57d6 · 00d57d6
1 parent c2b3cef
commit 00d57d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/swebench/harness/docker_build.py b/swebench/harness/docker_build.py
@@ -241,7 +241,7 @@ def get_env_configs_to_build(
                 # Remove the environment image if it was built after the base_image
                 for dep in find_dependent_images(client, test_spec.env_image_key):
                     # Remove instance images that depend on this environment image
-                    remove_image(client, dep.image_id, "quiet")
+                    remove_image(client, dep, "quiet")
                 remove_image(client, test_spec.env_image_key, "quiet")
                 image_exists = False
         except docker.errors.ImageNotFound: