Fixed render bug in RecordVideo wrapper (#128)

This commit is contained in:
lfwa
2022-11-16 13:21:26 +01:00
committed by GitHub
parent 92d3fa3e71
commit 55c02edc1d

View File

@@ -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."""