Update pd_grid

projectmesa · Sep 21, 2024 · a9d8a6a · a9d8a6a
1 parent 0b09f5c
commit a9d8a6a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/examples/pd_grid/pd_grid/model.py b/examples/pd_grid/pd_grid/model.py
@@ -6,11 +6,7 @@
 class PdGrid(mesa.Model):
     """Model class for iterated, spatial prisoner's dilemma model."""
 
-    schedule_types = {
-        "Sequential": mesa.time.BaseScheduler,
-        "Random": mesa.time.RandomActivation,
-        "Simultaneous": mesa.time.SimultaneousActivation,
-    }
+    activation_regimes = ["Sequential", "Random", "Simultaneous"]
 
     # This dictionary holds the payoff for this agent,
     # keyed on: (my_move, other_move)

diff --git a/examples/pd_grid/pd_grid/server.py b/examples/pd_grid/pd_grid/server.py
@@ -10,9 +10,9 @@
     "height": 50,
     "width": 50,
     "activation_order": mesa.visualization.Choice(
-        "Scheduler type",
+        "Activation regime",
         value="Random",
-        choices=list(PdGrid.schedule_types.keys()),
+        choices=PdGrid.activation_regimes,
     ),
 }