From 18cbc741cb7d7779b7993b8a69d12b10993f6b9b Mon Sep 17 00:00:00 2001
From: u229589 <christian.baumberger@sbb.ch>
Date: Mon, 23 Sep 2019 11:26:16 +0200
Subject: [PATCH] Refactoring: bugfix of return type
---
flatland/envs/observations.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index a9a9b9dc..3cb29a3f 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -640,7 +640,7 @@ class LocalObsForRailEnv(ObservationBuilder):
direction = np.identity(4)[agent.direction]
return local_rail_obs, obs_map_state, obs_other_agents_state, direction
- def get_many(self, handles: Optional[List[int]] = None) -> Dict[int, (np.ndarray, np.ndarray, np.ndarray, np.ndarray)]:
+ def get_many(self, handles: Optional[List[int]] = None) -> Dict[int, tuple]:
"""
Called whenever an observation has to be computed for the `env` environment, for each agent with handle
in the `handles` list.
--
GitLab