FastTreeObs working

run.py

@@ -9,6 +9,7 @@ from flatland.envs.predictions import ShortestPathPredictorForRailEnv
 from flatland.evaluators.client import FlatlandRemoteClient
 from flatland.evaluators.client import TimeoutException
 
+from utils.dead_lock_avoidance_agent import DeadLockAvoidanceAgent
 from utils.deadlock_check import check_if_all_blocked
 from utils.fast_tree_obs import FastTreeObs
 
@@ -81,6 +82,8 @@ while True:
     nb_agents = len(local_env.agents)
     max_nb_steps = local_env._max_episode_steps
 
+    policy = DeadLockAvoidanceAgent(local_env)
+
     tree_observation.set_env(local_env)
     tree_observation.reset()
     observation = tree_observation.get_many(list(range(nb_agents)))