models.py

from pydantic import Field

from openenv.core.env_server.types import Action, Observation


class NflPlaycallAction(Action):
    play_type: str = Field(..., description="One of: run, pass, play_action")


class NflPlaycallObservation(Observation):
    defensive_formation: str = Field(
        default="", description="Constant defensive formation"
    )
    yards_gained: float = Field(default=0.0, description="Yards gained on the play")