Fix clean_up_20 to use the correct puppet bot.

PiperOrigin-RevId: 650010429 Change-Id: Ie0765ee3e8b0017790ac152e17ea86b3878c3449
jagapiou · Jul 7, 2024 · bd6a547 · bd6a547
1 parent 76104ec
commit bd6a547
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/meltingpot/configs/bots/__init__.py b/meltingpot/configs/bots/__init__.py
@@ -1213,6 +1213,23 @@ def puppet(*,
             niceness_period=200,
         ),
     ),
+    clean_up__puppet_sanctioning_alternator_0=puppet(
+        substrate='clean_up',
+        model='puppet_3_goals_0',
+        roles=('default',),
+        puppeteer_builder=functools.partial(
+            clean_up.SanctionerAlternator,
+            cooperate_goal=_PUPPET_GOALS['clean_up_3_goals']['CLEAN'],
+            defect_goal=_PUPPET_GOALS['clean_up_3_goals']['EAT'],
+            sanction_goal=_PUPPET_GOALS['clean_up_3_goals']['SANCTION'],
+            num_others_cooperating_cumulant='NUM_OTHERS_WHO_CLEANED_THIS_STEP',
+            threshold=2,
+            recency_window=50,
+            steps_to_sanction_when_motivated=100,
+            alternating_steps=200,
+            nice=False,
+        ),
+    ),
     clean_up__puppet_sanctioning_alternator_nice_0=puppet(
         substrate='clean_up',
         model='puppet_3_goals_0',

diff --git a/meltingpot/configs/scenarios/__init__.py b/meltingpot/configs/scenarios/__init__.py
@@ -1462,7 +1462,7 @@ def __post_init__(self):
         is_focal=(False,) + (True,) * 6,  # Reversed is_focal for variety.
         bots_by_role={
             'default': {
-                'clean_up__puppet_sanctioning_alternator_nice_0',
+                'clean_up__puppet_sanctioning_alternator_0',
             },
         },
     ),