diff --git a/gym/envs/toy_text/frozen_lake.py b/gym/envs/toy_text/frozen_lake.py index 78b3e10ee25..a85b0e03bff 100644 --- a/gym/envs/toy_text/frozen_lake.py +++ b/gym/envs/toy_text/frozen_lake.py @@ -276,12 +276,11 @@ def _render_gui(self, desc, mode): ] self.elf_images = [pygame.image.load(f_name) for f_name in elfs] - board = pygame.Surface(self.window_size, flags=SRCALPHA) cell_width = self.window_size[0] // self.ncol cell_height = self.window_size[1] // self.nrow smaller_cell_scale = 0.6 - small_cell_w = smaller_cell_scale * cell_width - small_cell_h = smaller_cell_scale * cell_height + small_cell_w = int(smaller_cell_scale * cell_width) + small_cell_h = int(smaller_cell_scale * cell_height) # prepare images last_action = self.lastaction if self.lastaction is not None else 1 @@ -319,7 +318,7 @@ def _render_gui(self, desc, mode): else: self.window_surface.blit(ice_img, (rect[0], rect[1])) - pygame.draw.rect(board, (180, 200, 230), rect, 1) + pygame.draw.rect(self.window_surface, (180, 200, 230), rect, 1) # paint the elf bot_row, bot_col = self.s // self.ncol, self.s % self.ncol @@ -335,7 +334,6 @@ def _render_gui(self, desc, mode): elf_rect = self._center_small_rect(cell_rect, elf_img.get_size()) self.window_surface.blit(elf_img, elf_rect) - self.window_surface.blit(board, board.get_rect()) if mode == "human": pygame.event.pump() pygame.display.update()