Add memory limit to agent sandbox (rllm-org#101)

sinman-aisi · web-flow · commit ea121794fa45 · 2024-11-15T17:11:15.000Z
diff --git a/src/inspect_evals/cybench/challenges/avatar/compose.yaml b/src/inspect_evals/cybench/challenges/avatar/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-avatar-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/back_to_the_past/compose.yaml b/src/inspect_evals/cybench/challenges/back_to_the_past/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-back_to_the_past-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/chunky/compose.yaml b/src/inspect_evals/cybench/challenges/chunky/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   blog:
     image: aisiuk/evals-cybench-chunky-blog
     init: true
diff --git a/src/inspect_evals/cybench/challenges/crushing/compose.yaml b/src/inspect_evals/cybench/challenges/crushing/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/data_siege/compose.yaml b/src/inspect_evals/cybench/challenges/data_siege/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/delulu/compose.yaml b/src/inspect_evals/cybench/challenges/delulu/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-delulu-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/diffecient/compose.yaml b/src/inspect_evals/cybench/challenges/diffecient/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-diffecient-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/dynastic/compose.yaml b/src/inspect_evals/cybench/challenges/dynastic/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/eval_me/compose.yaml b/src/inspect_evals/cybench/challenges/eval_me/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/ezmaze/compose.yaml b/src/inspect_evals/cybench/challenges/ezmaze/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-ezmaze-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/failproof/compose.yaml b/src/inspect_evals/cybench/challenges/failproof/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-failproof-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/flecks_of_gold/compose.yaml b/src/inspect_evals/cybench/challenges/flecks_of_gold/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/frog_waf/compose.yaml b/src/inspect_evals/cybench/challenges/frog_waf/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-frog_waf-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/glacier_exchange/compose.yaml b/src/inspect_evals/cybench/challenges/glacier_exchange/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-glacier_exchange-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/it_has_begun/compose.yaml b/src/inspect_evals/cybench/challenges/it_has_begun/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/just_another_pickle_jail/compose.yaml b/src/inspect_evals/cybench/challenges/just_another_pickle_jail/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-just_another_pickle_jail-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/labyrinth_linguist/compose.yaml b/src/inspect_evals/cybench/challenges/labyrinth_linguist/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-labyrinth_linguist-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/locktalk/compose.yaml b/src/inspect_evals/cybench/challenges/locktalk/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-locktalk-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/lootstash/compose.yaml b/src/inspect_evals/cybench/challenges/lootstash/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/missingbits/compose.yaml b/src/inspect_evals/cybench/challenges/missingbits/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/motp/compose.yaml b/src/inspect_evals/cybench/challenges/motp/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-motp-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/network_tools/compose.yaml b/src/inspect_evals/cybench/challenges/network_tools/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-network_tools-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/noisier_crc/compose.yaml b/src/inspect_evals/cybench/challenges/noisier_crc/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-noisier_crc-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/noisy_crc/compose.yaml b/src/inspect_evals/cybench/challenges/noisy_crc/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-noisy_crc-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/packedaway/compose.yaml b/src/inspect_evals/cybench/challenges/packedaway/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/partial_tenacity/compose.yaml b/src/inspect_evals/cybench/challenges/partial_tenacity/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/permuted/compose.yaml b/src/inspect_evals/cybench/challenges/permuted/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/primary_knowledge/compose.yaml b/src/inspect_evals/cybench/challenges/primary_knowledge/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/randsubware/compose.yaml b/src/inspect_evals/cybench/challenges/randsubware/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-randsubware-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/robust_cbc/compose.yaml b/src/inspect_evals/cybench/challenges/robust_cbc/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-robust_cbc-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/rpgo/compose.yaml b/src/inspect_evals/cybench/challenges/rpgo/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/shuffled_aes/compose.yaml b/src/inspect_evals/cybench/challenges/shuffled_aes/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-shuffled_aes-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/skilift/compose.yaml b/src/inspect_evals/cybench/challenges/skilift/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-skilift-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/slcg/compose.yaml b/src/inspect_evals/cybench/challenges/slcg/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/sop/compose.yaml b/src/inspect_evals/cybench/challenges/sop/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/unbreakable/compose.yaml b/src/inspect_evals/cybench/challenges/unbreakable/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-unbreakable-victim
     init: true
diff --git a/src/inspect_evals/cybench/challenges/urgent/compose.yaml b/src/inspect_evals/cybench/challenges/urgent/compose.yaml
@@ -2,3 +2,4 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
diff --git a/src/inspect_evals/cybench/challenges/walking_to_the_sea_side/compose.yaml b/src/inspect_evals/cybench/challenges/walking_to_the_sea_side/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
     environment:
       alice: alice
       bob: bob
diff --git a/src/inspect_evals/cybench/challenges/were_pickle_phreaks_revenge/compose.yaml b/src/inspect_evals/cybench/challenges/were_pickle_phreaks_revenge/compose.yaml
@@ -2,6 +2,7 @@ services:
   default:
     image: aisiuk/evals-cybench-agent-sandbox
     init: true
+    mem_limit: 2g
   victim:
     image: aisiuk/evals-cybench-were_pickle_phreaks_revenge-victim
     init: true
