Fix multi attack sweep all_parallel

HumanCompatibleAI · Jul 15, 2022 · 2170308 · 2170308
1 parent c535767
commit 2170308
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/src/experiments/icml/multi_attack_sweep.py b/src/experiments/icml/multi_attack_sweep.py
@@ -13,24 +13,26 @@ def run_multi_atack_sweep(
     description: str = "",
     num_seeds: int = 3,
     parallel: bool = True,
+    all_parallel: bool = True,
     adversary_id: Optional[int] = None,
     victim_policy_name: Optional[str] = None,
     train_batch_size: Optional[int] = None,
     lr: Optional[float] = None,
 ):
     for victim_artifact in victim_artifacts:
         run_multi_atack(
-            env_config,
-            num_workers,
-            sweep_name,
-            victim_artifact,
-            description,
-            num_seeds,
-            parallel,
-            adversary_id,
-            victim_policy_name,
-            train_batch_size,
-            lr,
+            env_config=env_config,
+            num_workers=num_workers,
+            sweep_name=sweep_name,
+            victim_artifact=victim_artifact,
+            description=description,
+            num_seeds=num_seeds,
+            parallel=parallel,
+            all_parallel=all_parallel,
+            adversary_id=adversary_id,
+            victim_policy_name=victim_policy_name,
+            train_batch_size=train_batch_size,
+            lr=lr,
         )