@@ -552,18 +552,6 @@ async def run_build(
552552 default = None ,
553553 help = "Registry to tag and push images to (e.g., 'my-registry.com/myrepo')" ,
554554)
555- @click .option (
556- "--max-instances" ,
557- type = int ,
558- default = None ,
559- help = "Maximum number of instances to build (for swebench dataset)" ,
560- )
561- @click .option (
562- "--instance-ids" ,
563- multiple = True ,
564- default = None ,
565- help = "Specific instance IDs to build (for swebench dataset, can be specified multiple times)" ,
566- )
567555@click .option (
568556 "-v" ,
569557 "--verbose" ,
@@ -577,8 +565,6 @@ def main(
577565 cache_dir : str ,
578566 rebuild_existing : bool ,
579567 registry : str | None ,
580- max_instances : int | None ,
581- instance_ids : tuple [str , ...],
582568 verbose : bool ,
583569) -> None :
584570 """Build Docker images for datasets.
@@ -591,14 +577,8 @@ def main(
591577 # Build images for SWE-bench dataset
592578 prompt-siren-build-images --dataset swebench
593579
594- # Build images for browser-screenshot dataset
595- prompt-siren-build-images --dataset browser-screenshot
596-
597580 # Build images for all supported datasets
598581 prompt-siren-build-images --all-datasets
599-
600- # Build with a limited number of instances (for testing)
601- prompt-siren-build-images --dataset swebench --max-instances 5
602582 """
603583 # Configure logging
604584 log_level = logging .DEBUG if verbose else logging .INFO
@@ -620,24 +600,13 @@ def main(
620600 logger .error ("Must specify --dataset or --all-datasets" )
621601 sys .exit (1 )
622602
623- # Build config overrides for swebench
624- config_overrides : dict [str , dict [str , Any ]] = {}
625- if max_instances is not None or instance_ids :
626- swebench_overrides : dict [str , Any ] = {"cache_dir" : cache_dir }
627- if max_instances is not None :
628- swebench_overrides ["max_instances" ] = max_instances
629- if instance_ids :
630- swebench_overrides ["instance_ids" ] = list (instance_ids )
631- config_overrides ["swebench" ] = swebench_overrides
632-
633603 try :
634604 asyncio .run (
635605 run_build (
636606 datasets = datasets_to_build ,
637607 cache_dir = cache_dir ,
638608 rebuild_existing = rebuild_existing ,
639609 registry = registry ,
640- config_overrides = config_overrides ,
641610 )
642611 )
643612 except ExceptionGroup as eg :
0 commit comments