mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-02 06:16:32 +00:00
Fixed render bug in RecordVideo wrapper (#128)
This commit is contained in:
@@ -201,10 +201,7 @@ class RecordVideo(gym.Wrapper):
|
||||
else:
|
||||
return recorded_frames + super().render(*args, **kwargs)
|
||||
else:
|
||||
if self.recording:
|
||||
return self.video_recorder.last_frame
|
||||
else:
|
||||
return super().render(*args, **kwargs)
|
||||
return super().render(*args, **kwargs)
|
||||
|
||||
def close(self):
|
||||
"""Closes the wrapper then the video recorder."""
|
||||
|
Reference in New Issue
Block a user