From 0547a4364ad48c211fa6f1822f7072eb1cc63316 Mon Sep 17 00:00:00 2001
From: Michel Marti <michel.marti@t-systems.com>
Date: Tue, 21 Dec 2021 12:30:15 +0100
Subject: [PATCH] fixed bug: missing action_dict argument in check_deadlocks
and check_immediate_deadlocks
---
flatland/contrib/wrappers/flatland_wrappers.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/flatland/contrib/wrappers/flatland_wrappers.py b/flatland/contrib/wrappers/flatland_wrappers.py
index d4a2c324..ce3b2a23 100644
--- a/flatland/contrib/wrappers/flatland_wrappers.py
+++ b/flatland/contrib/wrappers/flatland_wrappers.py
@@ -258,6 +258,7 @@ class SkipNoChoiceCellsWrapper(RailEnvWrapper):
+
class DeadlockWrapper(RailEnvWrapper):
def __init__(self, env:RailEnv, deadlock_reward=-100) -> None:
super().__init__(env)
@@ -282,7 +283,7 @@ class DeadlockWrapper(RailEnvWrapper):
obs, rewards, dones, info = self.env.step(action_dict)
# compute new list of deadlocked agents (ids) after stepping the environment
- deadlocked_agents = self.deadlock_checker.check_deadlocks() # also stored in self.deadlocked_checker.deadlocked_agents
+ deadlocked_agents = self.deadlock_checker.check_deadlocks(action_dict) # also stored in self.deadlocked_checker.deadlocked_agents
deadlocked_agents_ids = [agent.handle for agent in deadlocked_agents]
# immediate deadlocked ids only used for prints
immediate_deadlocked_ids = [agent.handle for agent in self.deadlock_checker.immediate_deadlocked]
--
GitLab