fixed bug: missing action_dict argument in check_deadlocks and check_immediate_deadlocks

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]