Skip to content

Commit b5b967e

Browse files
committed
Remove old unneeded flags
1 parent b36e7b6 commit b5b967e

File tree

2 files changed

+1
-33
lines changed

2 files changed

+1
-33
lines changed

src/prompt_siren/build_images.py

Lines changed: 0 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -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:

src/prompt_siren/datasets/swebench_dataset/dataset.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,8 +107,7 @@ def get_image_build_specs(cls, config: SwebenchDatasetConfig) -> list[ImageBuild
107107

108108
if not instances:
109109
raise RuntimeError(
110-
"Cannot get image build specs: no SWE-bench instances match the filter criteria. "
111-
"Check instance_ids or max_instances in the dataset configuration."
110+
"Cannot get image build specs: no SWE-bench instances have known injection mappings."
112111
)
113112

114113
specs: list[ImageBuildSpec] = []

0 commit comments

Comments
 (0)