pantheonrl.common.agents

This module defines the standard Agent classes for PantheonRL.

It defines the following Agents:

  • The abstract base Agent class

  • The DummyAgent for wrapping SARL algorithms (not user-facing)

  • The SB3-specific OnPolicyAgent and OffPolicyAgent

  • The StaticPolicyAgent for pure inference of trained policies

It also defines the RecordingAgentWrapper to record transitions that an agent experiences.

Classes

Agent

Base class for all agents in multi-agent environments

DummyAgent

Agent wrapper for standard SARL algorithms assuming a gym interface.

OffPolicyAgent

Agent representing an off-policy learning algorithm (ex: DQN/SAC).

OnPolicyAgent

Agent representing an on-policy learning algorithm (ex: A2C/PPO).

RecordingAgentWrapper

Wrapper for an agent that records observation-action pairs.

StaticPolicyAgent

Agent representing a static (not learning) policy.