From 2afe31b9a576a27f96a55573bb226b1302774dd7 Mon Sep 17 00:00:00 2001
From: u229589 <christian.baumberger@sbb.ch>
Date: Fri, 27 Sep 2019 09:58:59 +0200
Subject: [PATCH] adjust return value for get and ge_many
---
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 30c0fabe..a08a5f73 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -58,7 +58,7 @@ class TreeObsForRailEnv(ObservationBuilder):
def reset(self):
self.location_has_target = {tuple(agent.target): 1 for agent in self.env.agents}
- def get_many(self, handles: Optional[List[int]] = None) -> Dict[int, List[int]]:
+ def get_many(self, handles: Optional[List[int]] = None) -> Dict[int, Node]:
"""
Called whenever an observation has to be computed for the `env` environment, for each agent with handle
in the `handles` list.
@@ -87,7 +87,7 @@ class TreeObsForRailEnv(ObservationBuilder):
observations[h] = self.get(h)
return observations
- def get(self, handle: int = 0) -> List[int]:
+ def get(self, handle: int = 0) -> Node:
"""
Computes the current observation for agent `handle` in env
--
GitLab