smarts.env.wrappers.recorder_wrapper module

class smarts.env.wrappers.recorder_wrapper.RecorderWrapper(*args: Any, **kwargs: Any)

A Wrapper that interacts the gym environment with the GifRecorder to record video step by step.

close()

Close the recorder by deleting the image folder and generate the gif file.

next_frame_id()

Get the id for next frame.

reset(**kwargs)

Reset the gym environment and restart recording.

start_recording()

Start the gif recorder and capture the first frame.

step(action)

Step the environment using the action and record the next frame.

stop_recording()

Stop recording.