From 00d57d6c9e10d18a027d3953be5223cfa2c4abd6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=98=93=E5=A4=A9=E8=8E=B2?=
 <91449279+yitianlian@users.noreply.github.com>
Date: Tue, 13 Aug 2024 20:58:33 +0800
Subject: [PATCH] fix bug when remove images

---
 swebench/harness/docker_build.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/swebench/harness/docker_build.py b/swebench/harness/docker_build.py
index 3f71ed84..d7bcf8ee 100644
--- 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: