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

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

Author: pre-commit-ci[bot]

gis/Traffic_Simulation/Server.py

@@ -7,6 +7,7 @@
 from mesa.visualization.ModularVisualization import ModularServer
 from flask import Flask
 
+
 def vehicle_portrayal(agent):
     if agent is None:
         return
@@ -16,11 +17,12 @@ def vehicle_portrayal(agent):
         "Color": "red",
         "Filled": "true",
         "Layer": 1,
-        "r": 3
+        "r": 3,
     }
 
     return portrayal
 
+
 def road_portrayal(road):
     return {
         "Shape": "line",
@@ -30,16 +32,19 @@ def road_portrayal(road):
         "coordinates": [(coord[1], coord[0]) for coord in road.shape.coords],
     }
 
+
 class RoadAgent(GeoAgent):
     def __init__(self, unique_id, model, shape, crs):
         super().__init__(unique_id, model, shape, crs)
 
+
 def create_road_agents(model):
     edges = ox.graph_to_gdfs(model.graph, nodes=False, edges=True)
     for i, row in edges.iterrows():
-        road = RoadAgent(i, model, row['geometry'], crs="epsg:4326")
+        road = RoadAgent(i, model, row["geometry"], crs="epsg:4326")
         model.space.add_agents(road)
 
+
 map_module = mg.visualization.MapModule(
     portrayal_method=vehicle_portrayal,
     view=[40.755, -73.987],  # Center view on a critical point
@@ -58,6 +63,7 @@ def create_road_agents(model):
     "steps": mesa.visualization.Slider("Simulation Steps", 10, 1, 100, 1),
 }
 
+
 class CustomModularServer(ModularServer):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
@@ -84,11 +90,9 @@ def simulate(self):
             self.model.step()
         self.model.running = False
 
+
 server = CustomModularServer(
-    TrafficModel,
-    [map_module],
-    "Traffic Simulation",
-    model_params
+    TrafficModel, [map_module], "Traffic Simulation", model_params
 )
 
 server.port = 8522  # Change to a different port if necessary
@@ -119,7 +123,7 @@ def simulate(self):
         L.tileLayer('https://{s}.tile.openstreetmap.org/{z}/{x}/{y}.png', {
             attribution: '&copy; <a href="https://www.openstreetmap.org/copyright">OpenStreetMap</a> contributors'
         }).addTo(map);
-        
+
         var simulate = function() {
             fetch('/simulate', { method: 'POST' });
         };

gis/Traffic_Simulation/model.py

@@ -13,13 +13,16 @@
     (40.752, -73.987),
 ]
 
+
 class VehicleAgent(GeoAgent):
     def __init__(self, unique_id, model, shape, start_node, end_node, crs):
         super().__init__(unique_id, model, shape, crs)
         self.start_node = start_node
         self.end_node = end_node
         try:
-            self.route = nx.shortest_path(model.graph, source=start_node, target=end_node, weight='length')
+            self.route = nx.shortest_path(
+                model.graph, source=start_node, target=end_node, weight="length"
+            )
         except nx.NetworkXNoPath:
             self.route = []  # Empty route if no path found
         self.current_step = 0
@@ -28,14 +31,17 @@ def step(self):
         if self.current_step < len(self.route) - 1:
             self.current_step += 1
             node = self.route[self.current_step]
-            self.shape = Point((self.model.graph.nodes[node]['x'], self.model.graph.nodes[node]['y']))
+            self.shape = Point(
+                (self.model.graph.nodes[node]["x"], self.model.graph.nodes[node]["y"])
+            )
+
 
 class TrafficModel(Model):
     def __init__(self, num_vehicles, north, south, east, west, steps):
         super().__init__()
         self.num_vehicles = num_vehicles
         self.steps = steps
-        self.graph = ox.graph_from_bbox(north, south, east, west, network_type='drive')
+        self.graph = ox.graph_from_bbox(north, south, east, west, network_type="drive")
         self.schedule = SimultaneousActivation(self)
         self.space = GeoSpace(crs="epsg:4326")
 
@@ -44,8 +50,12 @@ def __init__(self, num_vehicles, north, south, east, west, steps):
         for i in range(self.num_vehicles):
             start_node = random.choice(list(self.graph.nodes))
             end_node = random.choice(list(self.graph.nodes))
-            start_point = Point((self.graph.nodes[start_node]['x'], self.graph.nodes[start_node]['y']))
-            vehicle = VehicleAgent(i, self, start_point, start_node, end_node, crs="epsg:4326")
+            start_point = Point(
+                (self.graph.nodes[start_node]["x"], self.graph.nodes[start_node]["y"])
+            )
+            vehicle = VehicleAgent(
+                i, self, start_point, start_node, end_node, crs="epsg:4326"
+            )
             if vehicle.route:  # Only add vehicles with valid routes
                 self.space.add_agents(vehicle)
                 self.schedule.add(vehicle)
@@ -55,7 +65,9 @@ def simulate_congestion(self, critical_points):
             node = ox.distance.nearest_nodes(self.graph, X=point[1], Y=point[0])
             if node in self.graph.nodes:
                 for u, v, key, data in self.graph.edges(node, keys=True, data=True):
-                    data['length'] *= 5  # Simulating congestion by increasing travel time
+                    data["length"] *= (
+                        5  # Simulating congestion by increasing travel time
+                    )
 
     def step(self):
         self.schedule.step()