From 5ea881c2b4ef3aef382dbda448e226d46f17e1e2 Mon Sep 17 00:00:00 2001 From: Mario Jerez <72274387+mariojerez@users.noreply.github.com> Date: Sun, 24 Nov 2024 14:46:44 -0600 Subject: [PATCH] Fixed error in render function for CustomActionMaskedEnvironment Class (#1239) --- tutorials/CustomEnvironment/tutorial3_action_masking.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorials/CustomEnvironment/tutorial3_action_masking.py b/tutorials/CustomEnvironment/tutorial3_action_masking.py index 24676373f..c0dfe2170 100644 --- a/tutorials/CustomEnvironment/tutorial3_action_masking.py +++ b/tutorials/CustomEnvironment/tutorial3_action_masking.py @@ -193,7 +193,7 @@ def step(self, actions): def render(self): """Renders the environment.""" - grid = np.zeros((7, 7)) + grid = np.zeros((8, 8), dtype=object) grid[self.prisoner_y, self.prisoner_x] = "P" grid[self.guard_y, self.guard_x] = "G" grid[self.escape_y, self.escape_x] = "E"