SampleFactory
Integration for the SampleFactory framework with auto-reset and metrics forwarding.
Basic Usage
from pogema import pogema_v0, GridConfig
env = pogema_v0(GridConfig(
integration='SampleFactory',
auto_reset=True, # Auto-reset on episode end
num_agents=4,
size=8,
))
Features
- AutoResetWrapper: Automatically resets the environment when an episode ends
- IsMultiAgentWrapper: Adds
.num_agentsand.is_multiagentattributes - MetricsForwardingWrapper: Exposes metrics via
episode_extra_statsfor SampleFactory logging