From 4a4742a2a8ae8279724a63b9b552fdc9307c8ce7 Mon Sep 17 00:00:00 2001
From: u229589 <christian.baumberger@sbb.ch>
Date: Mon, 23 Sep 2019 11:49:10 +0200
Subject: [PATCH] Refactoring: bugfix of return type
---
flatland/envs/observations.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 3cb29a3f..712d2425 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -2,7 +2,7 @@
Collection of environment-specific ObservationBuilder.
"""
import pprint
-from typing import Optional, List, Dict
+from typing import Optional, List, Dict, T, Tuple
import numpy as np
@@ -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, tuple]:
+ def get_many(self, handles: Optional[List[int]] = None) -> Dict[int, Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray]]:
"""
Called whenever an observation has to be computed for the `env` environment, for each agent with handle
in the `handles` list.
--
GitLab