smarts.env.utils.record module

class smarts.env.utils.record.AgentCameraRGBRender(env: gymnasium.Env, max_agents: int = 1, enabled=True)[source]

Wraps the environment with rgb_array render mode capabilities.

render(mode='rgb_array', **kwargs)[source]

Render the given camera image in this environment.

reset(obs) → Any[source]

Record the reset of the environment.

step(obs, rewards, dones, infos)[source]

Record a step.