Source code for rog_rl.agent_state

from enum import Enum


class AgentState(Enum):
    # Susceptible, Exposed, Infectious, Symptomatic, Recovered/Dead
    SUSCEPTIBLE = 0  # *
    EXPOSED = 1  # o
    INFECTIOUS = 2  # O
    SYMPTOMATIC = 3  # S
    RECOVERED = 4  # R
    VACCINATED = 5  # V