From 90e9c1e0038d79ef44d070002ed0d31d88f7aa87 Mon Sep 17 00:00:00 2001
From: "S.P. Mohanty" <spmohanty91@gmail.com>
Date: Fri, 18 Oct 2019 19:06:57 +0200
Subject: [PATCH] Add a set_env on every reset
---
flatland/envs/rail_env.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index d5766416..4bd4f012 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -269,6 +269,12 @@ class RailEnv(Environment):
self.rail = rail
self.height, self.width = self.rail.grid.shape
+
+ # Do a new set_env call on the obs_builder to ensure
+ # that obs_builder specific instantiations are made according to the
+ # specifications of the current environment : like width, height, etc
+ self.obs_builder.set_env(self)
+
# NOTE : Ignore Validation on every reset. rail_generator should ensure that
# only valid grids are generated.
#
--
GitLab