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

for more information, see https://pre-commit.ci
DenysSvintsylo · Nov 3, 2024 · 47e72ad · 47e72ad
1 parent 315129c
commit 47e72ad
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 6 deletions.
diff --git a/examples/color_patches/color_patches/model.py b/examples/color_patches/color_patches/model.py
@@ -8,7 +8,8 @@ class ColorCell(mesa.Agent):
     """
     Represents a cell's opinion (visualized by a color)
     """
-    #test
+
+    # test
     OPINIONS = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
 
     def __init__(self, unique_id, model, initial_state, mutation_prob=0.1):
@@ -36,12 +37,18 @@ def determine_opinion(self):
             self.next_state = self.random.choice(ColorCell.OPINIONS)
         else:
             # Стандартний процес визначення "думки" на основі сусідів
-            neighbors = self.model.grid.get_neighbors(self.pos, moore=True, include_center=False)
+            neighbors = self.model.grid.get_neighbors(
+                self.pos, moore=True, include_center=False
+            )
             neighbors_opinion = Counter(neighbor.state for neighbor in neighbors)
             polled_opinions = neighbors_opinion.most_common()
 
             # Збираємо всі зв'язані думки (якщо є кілька з однаковою частотою)
-            tied_opinions = [opinion[0] for opinion in polled_opinions if opinion[1] == polled_opinions[0][1]]
+            tied_opinions = [
+                opinion[0]
+                for opinion in polled_opinions
+                if opinion[1] == polled_opinions[0][1]
+            ]
             self.next_state = self.random.choice(tied_opinions)
 
     def assume_opinion(self):
@@ -68,8 +75,10 @@ def __init__(self, width=20, height=20):
         self.schedule = mesa.time.RandomActivation(self)
 
         # Create agents
-        for (content, x, y) in self.grid.coord_iter():
-            initial_state = ColorCell.OPINIONS[self.random.randrange(0, len(ColorCell.OPINIONS))]
+        for content, x, y in self.grid.coord_iter():
+            initial_state = ColorCell.OPINIONS[
+                self.random.randrange(0, len(ColorCell.OPINIONS))
+            ]
             agent = ColorCell(self.next_id(), self, initial_state, mutation_prob=0.01)
             self.grid.place_agent(agent, (x, y))
             self.schedule.add(agent)

diff --git a/examples/color_patches/color_patches/server.py b/examples/color_patches/color_patches/server.py
@@ -50,7 +50,9 @@ def color_patch_draw(cell):
     portrayal = {"Shape": "rect", "w": 1, "h": 1, "Filled": "true", "Layer": 0}
     # Use cell.pos to get the (x, y) coordinates
     portrayal["x"], portrayal["y"] = cell.pos
-    portrayal["Color"] = _COLORS[cell.state]  # Make sure to use cell.state for the color
+    portrayal["Color"] = _COLORS[
+        cell.state
+    ]  # Make sure to use cell.state for the color
     return portrayal