bug fixed

utils/dead_lock_avoidance_agent.py

@@ -67,7 +67,7 @@ class DeadlockAvoidanceShortestDistanceWalker(ShortestDistanceWalker):
 
 
 class DeadLockAvoidanceAgent(Policy):
-    def _init__(self, env: RailEnv, action_size, show_debug_plot=False):
+    def __init__(self, env: RailEnv, action_size, show_debug_plot=False):
         self.env = env
         self.memory = None
         self.loss = 0

utils/fast_tree_obs.py

@@ -36,7 +36,7 @@ class FastTreeObs(ObservationBuilder):
         self.debug_render_path_list = []
         if self.env is not None:
             self.find_all_cell_where_agent_can_choose()
-            self.dead_lock_avoidance_agent = DeadLockAvoidanceAgent(self.env, 5)
+            self.dead_lock_avoidance_agent = DeadLockAvoidanceAgent(self.env, 5, False)
         else:
             self.dead_lock_avoidance_agent = None