From bc9dca3168fa6cd6abdd4939abcc6a56cf8d7120 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 3 Nov 2020 17:05:50 +0100
Subject: [PATCH] FastTreeObs working

---
 run.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/run.py b/run.py
index 09dc04d..7ad08cd 100644
--- a/run.py
+++ b/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)))
-- 
GitLab