smarts.env.gymnasium.wrappers.episode_logger module

class smarts.env.gymnasium.wrappers.episode_logger.EpisodeLogger(*args: Any, **kwargs: Any)[source]

Wraps a gym environment with simple episode logging capabilities.

close()[source]

Cap off the episode logging.

reset() → Tuple[Any, Dict[str, Any]][source]

Mark an episode reset for logging.

step(action: Any)[source]

Mark a step for logging.