pre-commit

LucasAlegre · May 7, 2024 · dca4792 · dca4792
1 parent ae9c293
commit dca4792
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/experiments/sarsa_resco.py b/experiments/sarsa_resco.py
@@ -49,7 +49,11 @@ def run(use_gui=False, episodes=50):
 
                 for ts_id in next_obs.keys():
                     agents[ts_id].learn(
-                        state=obs[ts_id], action=actions[ts_id], reward=r[ts_id], next_state=next_obs[ts_id], done=terminated[ts_id]
+                        state=obs[ts_id],
+                        action=actions[ts_id],
+                        reward=r[ts_id],
+                        next_state=next_obs[ts_id],
+                        done=terminated[ts_id],
                     )
                     obs[ts_id] = next_obs[ts_id]
 

diff --git a/sumo_rl/nets/4x4-Lucas/metrics/result_plot.py b/sumo_rl/nets/4x4-Lucas/metrics/result_plot.py
@@ -1,9 +1,7 @@
-import re
 import sys
 
 import matplotlib.pyplot as plt
 import numpy as np
-from scipy.interpolate import spline
 from sumolib.output import parse_fast