gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-31 10:09:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix the initialization of render mode set (#2938)

Browse Source
This commit is contained in:
KexianShen
2022-06-30 23:00:47 +08:00
committed by GitHub
parent 61a39f41bc
commit 2017f3ed9e
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gym/utils/renderer.py
View File

@@ -54,9 +54,9 @@ class Renderer:
This method should be usually called inside environment's step and reset method. This method should be usually called inside environment's step and reset method.
""" """
if self.mode is not None and self.mode not in SINGLE_RENDER: if self.mode is not None and self.mode not in self.single_render:
render_return = self.render(self.mode) render_return = self.render(self.mode)
if self.mode not in NO_RETURNS_RENDER: if self.mode not in self.no_returns_render:
self.render_list.append(render_return) self.render_list.append(render_return)
def get_renders(self) -> Optional[List]: def get_renders(self) -> Optional[List]:
@@ -64,9 +64,9 @@ class Renderer:
This method should be usually called in the environment's render method to retrieve the frames collected till this time step. This method should be usually called in the environment's render method to retrieve the frames collected till this time step.
""" """
if self.mode in SINGLE_RENDER: if self.mode in self.single_render:
return self.render(self.mode) return self.render(self.mode)
elif self.mode is not None and self.mode not in NO_RETURNS_RENDER: elif self.mode is not None and self.mode not in self.no_returns_render:
renders = self.render_list renders = self.render_list
self.render_list = [] self.render_list = []
return renders return renders