Merge branch 'main' of github.com:princeton-nlp/SWE-bench

swe-bench · Nov 22, 2024 · 691d333 · 691d333
2 parents 97050dd + 8b8dd10
commit 691d333
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/swebench/harness/prepare_images.py b/swebench/harness/prepare_images.py
@@ -12,7 +12,7 @@
 
 def filter_dataset_to_build(
         dataset: list,
-        instance_ids: list,
+        instance_ids: list | None,
         client: docker.DockerClient,
         force_rebuild: bool
     ):
@@ -28,6 +28,9 @@ def filter_dataset_to_build(
     # Get existing images
     existing_images = list_images(client)
     data_to_build = []
+
+    if instance_ids is None:
+        instance_ids = [instance[KEY_INSTANCE_ID] for instance in dataset]
 
     # Check if all instance IDs are in the dataset
     not_in_dataset = set(instance_ids).difference(set([instance[KEY_INSTANCE_ID] for instance in dataset]))