Fix empty instances error in prepare_images

princeton-nlp · Nov 22, 2024 · 8b8dd10 · 8b8dd10
1 parent b31927c
commit 8b8dd10
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]))