Skip to content

Commit 0eaa400

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent b859867 commit 0eaa400

File tree

4 files changed

+12
-9
lines changed

4 files changed

+12
-9
lines changed

examples/traffic_flow/app.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,4 +56,4 @@ def car_portrayal(agent):
5656
model_params=model_params,
5757
name="Traffic Flow",
5858
)
59-
page # noqa
59+
page # noqa

examples/traffic_flow/run.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from traffic_flow.model import TraficFlow
22

3+
34
def render_grid(model):
45
lines = []
56
for y in reversed(range(model.grid.height)):
@@ -10,11 +11,12 @@ def render_grid(model):
1011
lines.append("".join(row))
1112
return "\n".join(lines)
1213

14+
1315
if __name__ == "__main__":
1416
model = TraficFlow(width=20, height=5, n_cars=20, seed=1)
1517

1618
for t in range(10):
1719
print(f"tick {t}")
1820
print(render_grid(model))
1921
print()
20-
model.step()
22+
model.step()
Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,15 @@
11
from mesa import Agent
22

3+
34
class CarAgent(Agent):
45
def __init__(self, model):
56
super().__init__(model)
6-
7+
78
def move(self):
8-
x,y = self.pos
9+
x, y = self.pos
910
new_pos = ((x + 1) % self.model.grid.width, y)
1011
if self.model.grid.is_cell_empty(new_pos):
1112
self.model.grid.move_agent(self, new_pos)
1213

1314
def step(self):
1415
self.move()
15-

examples/traffic_flow/traffic_flow/model.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
11
from mesa import Model
22
from mesa.space import MultiGrid
3-
3+
44
from .agent import CarAgent
5-
5+
6+
67
class TraficFlow(Model):
78
def __init__(self, width=20, height=5, n_cars=10, seed=None):
89
super().__init__(seed=seed)
9-
10+
1011
self.grid = MultiGrid(width, height, torus=True)
1112

1213
for _ in range(n_cars):
@@ -20,4 +21,4 @@ def __init__(self, width=20, height=5, n_cars=10, seed=None):
2021
self.grid.place_agent(car, (x, y))
2122

2223
def step(self):
23-
self.agents.shuffle_do("step")
24+
self.agents.shuffle_do("step")

0 commit comments

Comments
 (0)