src/typing.py

import enum

from pydantic import BaseModel


class AgentType(enum.Enum):
    RANDOM = "random"
    MCTS = "mcts"
    SAC = "sac"


class RolloutPolicy(enum.Enum):
    RANDOM = "random"
    SAC = "sac"


class CpuConfig(BaseModel):
    agent_type: AgentType
    simulations: int | None = None
    rollout_policy: RolloutPolicy | None = None

    # def __format__(self, __format_spec: str) -> str:
    #     raise ValueError(f"__format__ not implemented for {self.__class__.__name__}")
    #     return super().__format__(__format_spec)