Mesa 3.1.0__tar.gz → 3.1.1__tar.gz

{mesa-3.1.0 → mesa-3.1.1}/HISTORY.md

@@ -1,6 +1,35 @@
 ---
 title: Release History
 ---
+# 3.1.1 (2024-12-14)
+## Highlights
+Mesa 3.1.1 is a maintenance release that includes visualization improvements and documentation updates. The key enhancement is the addition of an interactive play interval control to the visualization interface, allowing users to dynamically adjust simulation speed between 1ms and 500ms through a slider in the Controls panel.
+
+Several example models were updated to use Mesa 3.1's recommended practices, particularly the `create_agents()` method for more efficient agent creation and NumPy's `rng.integers()` for random number generation. The Sugarscape example was modernized to use PropertyLayers.
+
+Bug fixes include improvements to PropertyLayer visualization and a correction to the Schelling model's neighbor similarity calculation. The tutorials were also updated to reflect current best practices in Mesa 3.1.
+
+## What's Changed
+### 🎉 New features added
+* Add Interactive Play Interval Control to Mesa Visualization by @AdamZh0u in https://github.com/projectmesa/mesa/pull/2540
+### 🐛 Bugs fixed
+* bug fixes for draw_property_layers by @quaquel in https://github.com/projectmesa/mesa/pull/2548
+### 🔍 Examples updated
+* Wolf-sheep to use `create_agent` by @quaquel in https://github.com/projectmesa/mesa/pull/2543
+* Shift sugarscape example to using create_agent by @quaquel in https://github.com/projectmesa/mesa/pull/2544
+* Fix: Schelling Model Neighbor Similarity Calculation by @Sahil-Chhoker in https://github.com/projectmesa/mesa/pull/2518
+* Change pd_grid example to use create_agents by @quaquel in https://github.com/projectmesa/mesa/pull/2545
+* Switch sugarscape to using property layers by @quaquel in https://github.com/projectmesa/mesa/pull/2546
+### 📜 Documentation improvements
+* Updated docs and check_model param by @nissu99 in https://github.com/projectmesa/mesa/pull/2510
+* Update tutorials to use `create_agents` and `rng.integers` by @DarshPareek in https://github.com/projectmesa/mesa/pull/2541
+
+## New Contributors
+* @nissu99 made their first contribution in https://github.com/projectmesa/mesa/pull/2510
+* @DarshPareek made their first contribution in https://github.com/projectmesa/mesa/pull/2541
+
+**Full Changelog**: https://github.com/projectmesa/mesa/compare/v3.1.0...3.1.1
+
 # 3.1.0 (2024-12-04)
 ## Highlights
 With Mesa 3.1.0 we're back on our regular release schedule after the big Mesa 3.0 release, with some exciting new features.

{mesa-3.1.0 → mesa-3.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: Mesa
-Version: 3.1.0
+Version: 3.1.1
 Summary: Agent-based modeling (ABM) in Python
 Project-URL: homepage, https://github.com/projectmesa/mesa
 Project-URL: repository, https://github.com/projectmesa/mesa

{mesa-3.1.0 → mesa-3.1.1}/docs/getting_started.md

@@ -180,6 +180,16 @@ Mesa now uses a new browser-based visualization system called SolaraViz. This al
 
 > **Note:** SolaraViz is experimental and still in active development for Mesa 3.0. While we attempt to minimize them, there might be API breaking changes between Mesa 3.0 and 3.1. There won't be breaking changes between Mesa 3.0.x patch releases.
 
+> **Note:** SolaraViz instantiates new models using `**model_parameters.value`, so all model inputs must be keyword arguments.
+
+Ensure your model's `__init__` method accepts keyword arguments matching the `model_params` keys.
+
+```python
+class MyModel(Model):
+    def __init__(self, n_agents=10, seed=None):
+        super().__init__(seed=seed)
+        # Initialize the model with N agents
+
 The core functionality for building your own visualizations resides in the [`mesa.visualization`](apis/visualization) namespace
 
 Here's a basic example of how to set up a visualization:

{mesa-3.1.0 → mesa-3.1.1}/docs/migration_guide.md

@@ -259,6 +259,17 @@ the import from mesa.experimental. Otherwise here is a list of things you need t
 
 Previously SolaraViz was initialized by providing a `model_cls` and a `model_params`. This has changed to expect a model instance `model`. You can still provide (user-settable) `model_params`, but only if users should be able to change them. It is now also possible to pass in a "reactive model" by first calling `model = solara.reactive(model)`. This is useful for notebook environments. It allows you to pass the model to the SolaraViz Module, but continue to use the model. For example calling `model.value.step()` (notice the extra .value) will automatically update the plots. This currently only automatically works for the step method, you can force visualization updates by calling `model.value.force_update()`.
 
+### Model Initialization with Keyword Arguments
+
+With the introduction of SolaraViz in Mesa 3.0, models are now instantiated using `**model_parameters.value`. This means all inputs for initializing a new model must be keyword arguments. Ensure your model's `__init__` method accepts keyword arguments matching the keys in `model_params`.
+
+```python
+class MyModel(mesa.Model):
+    def __init__(self, n_agents=10, seed=None):
+        super().__init__(seed=seed)
+        # Initialize the model with N agents
+```
+
 #### Default space visualization
 
 Previously we included a default space drawer that you could configure with an `agent_portrayal` function. You now have to explicitly create a space drawer with the `agent_portrayal` function

{mesa-3.1.0 → mesa-3.1.1}/docs/tutorials/MoneyModel.py

@@ -62,12 +62,13 @@ class MoneyModel(mesa.Model):
         self.grid = mesa.space.MultiGrid(width, height, True)
 
         # Create agents
-        for _ in range(self.num_agents):
-            a = MoneyAgent(self)
+        agents = MoneyAgent.create_agents(model=self, n=n)
+        # Create x and y positions for agents
+        x = self.rng.integers(0, self.grid.width, size=(n,))
+        y = self.rng.integers(0, self.grid.height, size=(n,))
+        for a, i, j in zip(agents, x, y):
             # Add the agent to a random grid cell
-            x = self.random.randrange(self.grid.width)
-            y = self.random.randrange(self.grid.height)
-            self.grid.place_agent(a, (x, y))
+            self.grid.place_agent(a, (i, j))
 
         self.datacollector = mesa.DataCollector(
             model_reporters={"Gini": compute_gini}, agent_reporters={"Wealth": "wealth"}

{mesa-3.1.0 → mesa-3.1.1}/docs/tutorials/intro_tutorial.ipynb

@@ -246,8 +246,7 @@
     "        super().__init__(seed=seed)\n",
     "        self.num_agents = n\n",
     "        # Create agents\n",
-    "        for _ in range(self.num_agents):\n",
-    "            a = MoneyAgent(self)"
+    "        MoneyAgent.create_agents(model=self, n=n)"
    ]
   },
   {
@@ -295,11 +294,8 @@
     "        super().__init__(seed=seed)\n",
     "        self.num_agents = n\n",
     "\n",
-    "        # Create agents\n",
-    "        for _ in range(self.num_agents):\n",
-    "            a = MoneyAgent(\n",
-    "                self\n",
-    "            )  # This calls the agent class parameter n number of times\n",
+    "        # Create n agents\n",
+    "        MoneyAgent.create_agents(model=self, n=n)\n",
     "\n",
     "    def step(self):\n",
     "        \"\"\"Advance the model by one step.\"\"\"\n",
@@ -478,10 +474,7 @@
     "        self.num_agents = n\n",
     "\n",
     "        # Create agents\n",
-    "        for _ in range(self.num_agents):\n",
-    "            a = MoneyAgent(\n",
-    "                self\n",
-    "            )  # This calls the agent class parameter n number of times\n",
+    "        MoneyAgent.create_agents(model=self, n=n)\n",
     "\n",
     "    def step(self):\n",
     "        \"\"\"Advance the model by one step.\"\"\"\n",
@@ -620,13 +613,13 @@
     "        self.grid = mesa.space.MultiGrid(width, height, True)\n",
     "\n",
     "        # Create agents\n",
-    "        for _ in range(self.num_agents):\n",
-    "            a = MoneyAgent(self)\n",
-    "\n",
+    "        agents = MoneyAgent.create_agents(model=self, n=n)\n",
+    "        # Create x and y positions for agents\n",
+    "        x = self.rng.randint(0, self.grid.width, size=(n,))\n",
+    "        y = self.rng.randint(0, self.grid.height, size=(n,))\n",
+    "        for a, i, j in zip(agents, x, y):\n",
     "            # Add the agent to a random grid cell\n",
-    "            x = self.random.randrange(self.grid.width)\n",
-    "            y = self.random.randrange(self.grid.height)\n",
-    "            self.grid.place_agent(a, (x, y))"
+    "            self.grid.place_agent(a, (i, j))"
    ]
   },
   {
@@ -724,12 +717,13 @@
     "        self.grid = mesa.space.MultiGrid(width, height, True)\n",
     "\n",
     "        # Create agents\n",
-    "        for _ in range(self.num_agents):\n",
-    "            a = MoneyAgent(self)\n",
+    "        agents = MoneyAgent.create_agents(model=self, n=n)\n",
+    "        # Create x and y coordinates for agents\n",
+    "        x = self.rng.integers(0, self.grid.width, size=(n,))\n",
+    "        y = self.rng.integers(0, self.grid.height, size=(n,))\n",
+    "        for a, i, j in zip(agents, x, y):\n",
     "            # Add the agent to a random grid cell\n",
-    "            x = self.random.randrange(self.grid.width)\n",
-    "            y = self.random.randrange(self.grid.height)\n",
-    "            self.grid.place_agent(a, (x, y))\n",
+    "            self.grid.place_agent(a, (i, j))\n",
     "\n",
     "    def step(self):\n",
     "        self.agents.shuffle_do(\"move\")\n",
@@ -852,7 +846,7 @@
     "            other = self.random.choice(cellmates)\n",
     "            other.wealth += 1\n",
     "            self.wealth -= 1\n",
-    "            \n",
+    "\n",
     "\n",
     "class MoneyModel(mesa.Model):\n",
     "    \"\"\"A model with some number of agents.\"\"\"\n",
@@ -868,12 +862,13 @@
     "        )\n",
     "\n",
     "        # Create agents\n",
-    "        for _ in range(self.num_agents):\n",
-    "            a = MoneyAgent(self)\n",
+    "        agents = MoneyAgent.create_agents(model=self, n=n)\n",
+    "        # Create x and y positions for agents\n",
+    "        x = self.rng.integers(0, self.grid.width, size=(n,))\n",
+    "        y = self.rng.integers(0, self.grid.height, size=(n,))\n",
+    "        for a, i, j in zip(agents, x, y):\n",
     "            # Add the agent to a random grid cell\n",
-    "            x = self.random.randrange(self.grid.width)\n",
-    "            y = self.random.randrange(self.grid.height)\n",
-    "            self.grid.place_agent(a, (x, y))\n",
+    "            self.grid.place_agent(a, (i, j))\n",
     "\n",
     "    def step(self):\n",
     "        self.datacollector.collect(self)\n",
@@ -1137,12 +1132,13 @@
     "        self.grid = mesa.space.MultiGrid(width, height, True)\n",
     "\n",
     "        # Create agents\n",
-    "        for _ in range(self.num_agents):\n",
-    "            a = MoneyAgent(self)\n",
+    "        agents = MoneyAgent.create_agents(model=self, n=n)\n",
+    "        # Create x and y positions for agents\n",
+    "        x = self.rng.integers(0, self.grid.width, size=(n,))\n",
+    "        y = self.rng.integers(0, self.grid.height, size=(n,))\n",
+    "        for a, i, j in zip(agents, x, y):\n",
     "            # Add the agent to a random grid cell\n",
-    "            x = self.random.randrange(self.grid.width)\n",
-    "            y = self.random.randrange(self.grid.height)\n",
-    "            self.grid.place_agent(a, (x, y))\n",
+    "            self.grid.place_agent(a, (i, j))\n",
     "\n",
     "        self.datacollector = mesa.DataCollector(\n",
     "            model_reporters={\"Gini\": compute_gini}, agent_reporters={\"Wealth\": \"wealth\"}\n",
@@ -1234,8 +1230,7 @@
     "        self.num_agents = n\n",
     "\n",
     "        # Create agents\n",
-    "        for _ in range(self.num_agents):\n",
-    "            a = MoneyAgent(self)\n",
+    "        MoneyAgent.create_agents(model=self, n=n)\n",
     "\n",
     "        self.datacollector = mesa.DataCollector(\n",
     "            model_reporters={\"Gini\": compute_gini}, agent_reporters={\"Wealth\": \"wealth\"}\n",
@@ -1320,8 +1315,7 @@
     "        ethnicities = [\"Green\", \"Blue\", \"Mixed\"]\n",
     "\n",
     "        # Create agents\n",
-    "        for _ in range(self.num_agents):\n",
-    "            a = MoneyAgent(self, self.random.choice(ethnicities))\n",
+    "        MoneyAgent.create_agents(model=self, n=n, ethnicity=self.random.choice(ethnicities))\n",
     "\n",
     "        self.datacollector = mesa.DataCollector(\n",
     "            model_reporters={\"Gini\": compute_gini},\n",
@@ -1406,12 +1400,13 @@
     "        self.running = True\n",
     "\n",
     "        # Create agents\n",
-    "        for _ in range(self.num_agents):\n",
-    "            a = MoneyAgent(self)\n",
+    "        agents = MoneyAgent.create_agents(model=self, n=n)\n",
+    "        # Create x and y positions for agents\n",
+    "        x = self.rng.integers(0, self.grid.width, size=(n,))\n",
+    "        y = self.rng.integers(0, self.grid.height, size=(n,))\n",
+    "        for a, i, j in zip(agents, x, y):\n",
     "            # Add the agent to a random grid cell\n",
-    "            x = self.random.randrange(self.grid.width)\n",
-    "            y = self.random.randrange(self.grid.height)\n",
-    "            self.grid.place_agent(a, (x, y))\n",
+    "            self.grid.place_agent(a, (i, j))\n",
     "\n",
     "        self.datacollector = mesa.DataCollector(\n",
     "            model_reporters={\"Gini\": compute_gini},\n",
@@ -1902,7 +1897,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.5"
+   "version": "3.12.3"
   },
   "widgets": {
    "state": {},

{mesa-3.1.0 → mesa-3.1.1}/docs/tutorials/visualization_tutorial.ipynb

@@ -3,7 +3,9 @@
   {
    "cell_type": "markdown",
    "metadata": {},
-   "source": "# Visualization Tutorial"
+   "source": [
+    "# Visualization Tutorial"
+   ]
   },
   {
    "cell_type": "markdown",
@@ -51,10 +53,10 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "import mesa\n",
     "print(f\"Mesa version: {mesa.__version__}\")\n",
@@ -66,10 +68,10 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "def agent_portrayal(agent):\n",
     "    return {\n",
@@ -79,15 +81,17 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "markdown",
-   "source": "In addition to the portrayal method, we instantiate the model parameters, some of which are modifiable by user inputs. In this case, the number of agents, N, is specified as a slider of integers."
+   "metadata": {},
+   "source": [
+    "In addition to the portrayal method, we instantiate the model parameters, some of which are modifiable by user inputs. In this case, the number of agents, N, is specified as a slider of integers."
+   ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "model_params = {\n",
     "    \"n\": {\n",
@@ -104,8 +108,8 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "markdown",
+   "metadata": {},
    "source": [
     "Next, we instantiate the visualization object which (by default) displays the grid containing the agents, and timeseries of values computed by the model's data collector. In this example, we specify the Gini coefficient.\n",
     "\n",
@@ -118,13 +122,13 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "# Create initial model instance\n",
-    "model1 = MoneyModel(50, 10, 10)\n",
+    "model1 = MoneyModel(n=50, width=10, height=10) #keyword arguments\n",
     "\n",
     "SpaceGraph = make_space_component(agent_portrayal)\n",
     "GiniPlot = make_plot_component(\"Gini\")\n",
@@ -140,8 +144,8 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "markdown",
+   "metadata": {},
    "source": [
     "## Part 2 - Dynamic Agent Representation \n",
     "\n",
@@ -155,10 +159,10 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "import mesa\n",
     "print(f\"Mesa version: {mesa.__version__}\")\n",
@@ -169,10 +173,10 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "def agent_portrayal(agent):\n",
     "    size = 10\n",
@@ -197,13 +201,13 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "# Create initial model instance\n",
-    "model1 = MoneyModel(50, 10, 10)\n",
+    "model = MoneyModel(n=50, width=10, height=10)\n",
     "\n",
     "SpaceGraph = make_space_component(agent_portrayal)\n",
     "GiniPlot = make_plot_component(\"Gini\")\n",
@@ -219,8 +223,8 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "markdown",
+   "metadata": {},
    "source": [
     "## Part 3 - Custom Components \n",
     "\n",
@@ -236,10 +240,10 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "import mesa\n",
     "print(f\"Mesa version: {mesa.__version__}\")\n",
@@ -253,10 +257,10 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "def agent_portrayal(agent):\n",
     "    size = 10\n",
@@ -281,15 +285,17 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "markdown",
-   "source": "Next, we update our solara frontend to use this new component"
+   "metadata": {},
+   "source": [
+    "Next, we update our solara frontend to use this new component"
+   ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "@solara.component\n",
     "def Histogram(model):\n",
@@ -306,13 +312,13 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "metadata": {},
+   "outputs": [],
    "source": [
     "# Create initial model instance\n",
-    "model1 = MoneyModel(50, 10, 10)\n",
+    "model = MoneyModel(n=50, width=10, height=10)\n",
     "\n",
     "SpaceGraph = make_space_component(agent_portrayal)\n",
     "GiniPlot = make_plot_component(\"Gini\")"
@@ -320,42 +326,42 @@
   },
   {
    "cell_type": "code",
+   "execution_count": 12,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2024-10-29T19:38:49.471838Z",
      "start_time": "2024-10-29T19:38:47.897295Z"
     }
    },
-   "source": [
-    "page = SolaraViz(\n",
-    "    model1,\n",
-    "    components=[SpaceGraph, GiniPlot, Histogram],\n",
-    "    model_params=model_params,\n",
-    "    name=\"Boltzmann Wealth Model\",\n",
-    ")\n",
-    "# This is required to render the visualization in the Jupyter notebook\n",
-    "page"
-   ],
    "outputs": [
     {
      "data": {
-      "text/plain": [
-       "Cannot show ipywidgets in text"
-      ],
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "bc71b89ee5684038a194eee4c36f4a4c",
+       "version_major": 2,
+       "version_minor": 0
+      },
       "text/html": [
        "Cannot show widget. You probably want to rerun the code cell above (<i>Click in the code cell, and press Shift+Enter <kbd>⇧</kbd>+<kbd>↩</kbd></i>)."
       ],
-      "application/vnd.jupyter.widget-view+json": {
-       "version_major": 2,
-       "version_minor": 0,
-       "model_id": "bc71b89ee5684038a194eee4c36f4a4c"
-      }
+      "text/plain": [
+       "Cannot show ipywidgets in text"
+      ]
      },
      "metadata": {},
      "output_type": "display_data"
     }
    ],
-   "execution_count": 12
+   "source": [
+    "page = SolaraViz(\n",
+    "    model,\n",
+    "    components=[SpaceGraph, GiniPlot, Histogram],\n",
+    "    model_params=model_params,\n",
+    "    name=\"Boltzmann Wealth Model\",\n",
+    ")\n",
+    "# This is required to render the visualization in the Jupyter notebook\n",
+    "page"
+   ]
   },
   {
    "cell_type": "markdown",
@@ -366,35 +372,35 @@
   },
   {
    "cell_type": "code",
+   "execution_count": 13,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2024-10-29T19:38:49.505725Z",
      "start_time": "2024-10-29T19:38:49.472599Z"
     }
    },
-   "source": [
-    "Histogram(model1)"
-   ],
    "outputs": [
     {
      "data": {
-      "text/plain": [
-       "Cannot show ipywidgets in text"
-      ],
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "0491f167a1434a92b78535078bd082a8",
+       "version_major": 2,
+       "version_minor": 0
+      },
       "text/html": [
        "Cannot show widget. You probably want to rerun the code cell above (<i>Click in the code cell, and press Shift+Enter <kbd>⇧</kbd>+<kbd>↩</kbd></i>)."
       ],
-      "application/vnd.jupyter.widget-view+json": {
-       "version_major": 2,
-       "version_minor": 0,
-       "model_id": "0491f167a1434a92b78535078bd082a8"
-      }
+      "text/plain": [
+       "Cannot show ipywidgets in text"
+      ]
      },
      "metadata": {},
      "output_type": "display_data"
     }
    ],
-   "execution_count": 13
+   "source": [
+    "Histogram(model)"
+   ]
   },
   {
    "cell_type": "markdown",

{mesa-3.1.0 → mesa-3.1.1}/mesa/__init__.py

@@ -22,7 +22,7 @@ __all__ = [
 ]
 
 __title__ = "mesa"
-__version__ = "3.1.0"
+__version__ = "3.1.1"
 __license__ = "Apache 2.0"
 _this_year = datetime.datetime.now(tz=datetime.UTC).date().year
 __copyright__ = f"Copyright {_this_year} Project Mesa Team"

{mesa-3.1.0 → mesa-3.1.1}/mesa/examples/advanced/pd_grid/agents.py

@@ -4,7 +4,7 @@ from mesa.experimental.cell_space import CellAgent
 class PDAgent(CellAgent):
     """Agent member of the iterated, spatial prisoner's dilemma model."""
 
-    def __init__(self, model, starting_move=None):
+    def __init__(self, model, starting_move=None, cell=None):
         """
         Create a new Prisoner's Dilemma agent.
 
@@ -15,6 +15,7 @@ class PDAgent(CellAgent):
         """
         super().__init__(model)
         self.score = 0
+        self.cell = cell
         if starting_move:
             self.move = starting_move
         else:

{mesa-3.1.0 → mesa-3.1.1}/mesa/examples/advanced/pd_grid/model.py

@@ -32,11 +32,9 @@ class PdGrid(mesa.Model):
         if payoffs is not None:
             self.payoff = payoffs
 
-        # Create agents
-        for x in range(width):
-            for y in range(height):
-                agent = PDAgent(self)
-                agent.cell = self.grid[(x, y)]
+        PDAgent.create_agents(
+            self, len(self.grid.all_cells.cells), cell=self.grid.all_cells.cells
+        )
 
         self.datacollector = mesa.DataCollector(
             {