Skip to content

Commit 9707791

Browse files
authored
Limit SWE-Bench to a single core per sample (rllm-org#114)
* restrict to 1 cpu * try cpu_count * use deploy instead
1 parent f7fbf6a commit 9707791

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/inspect_evals/swe_bench/swe_bench.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,11 @@ def get_compose_file(instance_id: str, ids_to_docker_image: dict[str, str]) -> s
185185
image: {image_name}
186186
command: "sleep infinity"
187187
working_dir: /testbed
188-
x-local: true"""
188+
x-local: true
189+
deploy:
190+
resources:
191+
limits:
192+
cpus: '1'"""
189193
)
190194

191195
return str(image_compose_file)

0 commit comments

Comments
 (0)