From af2013fe262de8cc07ae714dcced6a80e35f8ee1 Mon Sep 17 00:00:00 2001
From: Michel Marti <michel.marti@t-systems.com>
Date: Tue, 21 Dec 2021 12:41:42 +0100
Subject: [PATCH] added import of Deadlock_Checker

---
 flatland/contrib/wrappers/flatland_wrappers.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/flatland/contrib/wrappers/flatland_wrappers.py b/flatland/contrib/wrappers/flatland_wrappers.py
index ce3b2a23..500f3d2e 100644
--- a/flatland/contrib/wrappers/flatland_wrappers.py
+++ b/flatland/contrib/wrappers/flatland_wrappers.py
@@ -8,6 +8,7 @@ from flatland.utils.rendertools import RenderTool, AgentRenderVariant
 from flatland.envs.agent_utils import EnvAgent
 from flatland.envs.step_utils.states import TrainState
 from flatland.envs.rail_env import RailEnv, RailEnvActions
+from flatland.contrib.utils.deadlock_checker import Deadlock_Checker
 
 
 def possible_actions_sorted_by_distance(env: RailEnv, handle: int):
-- 
GitLab