[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

examples/needs_based_wolf_sheep/agents.py

@@ -1,4 +1,3 @@
-import mesa
 from mesa.discrete_space import CellAgent
 
 
@@ -18,13 +17,10 @@ def update_drives(self):
         self.fear = max(0.0, self.fear - 0.1)
 
     def move(self):
-        self.cell = self.random.choice(
-            list(self.cell.connections.values())
-        )
+        self.cell = self.random.choice(list(self.cell.connections.values()))
 
 
 class NeedsBasedWolf(NeedsBasedAnimal):
-
     def step(self):
         self.move()
         self.energy -= 1
@@ -35,34 +31,28 @@ def step(self):
             return
 
         if self.hunger > 0.6:
-            sheep = [
-                a for a in self.cell.agents
-                if isinstance(a, NeedsBasedSheep)
-            ]
+            sheep = [a for a in self.cell.agents if isinstance(a, NeedsBasedSheep)]
             if sheep:
                 prey = self.random.choice(sheep)
                 self.energy += 4
                 self.hunger = 0.0
                 prey.remove()
                 return
 
-        if (self.energy > 20 and self.fear < 0.3
-                and self.hunger < 0.4):
+        if self.energy > 20 and self.fear < 0.3 and self.hunger < 0.4:
             if self.random.random() < self.model.wolf_reproduce:
                 self.energy //= 2
                 NeedsBasedWolf(self.model, self.energy)
 
 
 class NeedsBasedSheep(NeedsBasedAnimal):
-
     def step(self):
         self.move()
         self.energy -= 1
         self.update_drives()
 
         wolves_nearby = sum(
-            1 for a in self.cell.agents
-            if isinstance(a, NeedsBasedWolf)
+            1 for a in self.cell.agents if isinstance(a, NeedsBasedWolf)
         )
         if wolves_nearby > 0:
             self.fear = min(1.0, self.fear + 0.5)
@@ -78,8 +68,7 @@ def step(self):
                 self.cell.grass = False
                 return
 
-        if (self.energy > 6 and self.fear < 0.2
-                and self.hunger < 0.3):
+        if self.energy > 6 and self.fear < 0.2 and self.hunger < 0.3:
             if self.random.random() < self.model.sheep_reproduce:
                 self.energy //= 2
-                NeedsBasedSheep(self.model, self.energy)
+                NeedsBasedSheep(self.model, self.energy)

examples/needs_based_wolf_sheep/model.py

@@ -1,6 +1,6 @@
 import mesa
+from agents import NeedsBasedSheep, NeedsBasedWolf
 from mesa.discrete_space import OrthogonalMooreGrid
-from agents import NeedsBasedWolf, NeedsBasedSheep
 
 
 class GrassPatch(mesa.Agent):
@@ -36,8 +36,7 @@ def __init__(
         self.grass_regrowth_rate = grass_regrowth_rate
 
         self.grid = OrthogonalMooreGrid(
-            (width, height), torus=True, capacity=None,
-            random=self.random
+            (width, height), torus=True, capacity=None, random=self.random
         )
 
         # Place grass patches
@@ -59,12 +58,8 @@ def __init__(
 
         self.datacollector = mesa.DataCollector(
             model_reporters={
-                "Wolves": lambda m: len(
-                    m.agents_by_type[NeedsBasedWolf]
-                ),
-                "Sheep": lambda m: len(
-                    m.agents_by_type[NeedsBasedSheep]
-                ),
+                "Wolves": lambda m: len(m.agents_by_type[NeedsBasedWolf]),
+                "Sheep": lambda m: len(m.agents_by_type[NeedsBasedSheep]),
             }
         )
 
@@ -79,8 +74,8 @@ def step(self):
         model.step()
         data = model.datacollector.get_model_vars_dataframe()
         print(
-            f"Step {i+1:3d}: "
+            f"Step {i + 1:3d}: "
             f"Wolves={int(data['Wolves'].iloc[-1]):3d}, "
             f"Sheep={int(data['Sheep'].iloc[-1]):3d}"
         )
-    print("Done — model ran 100 steps successfully.")
+    print("Done — model ran 100 steps successfully.")