models.py

"""Top-level Action / Observation / State re-exports for OpenEnv.

The OpenEnv official template puts typed Pydantic models at the package
root as ``models.py``. Our implementation lives in
:mod:`hospital_env.openenv_types` — this module is just a thin
re-export so clients can do::

    from models import HospitalAction, HospitalObservation, HospitalState

or the conventional unqualified names::

    from models import Action, Observation, State
"""
from hospital_env.openenv_types import (
    HospitalAction,
    HospitalObservation,
    HospitalState,
)

# Conventional OpenEnv unqualified aliases.
Action = HospitalAction
Observation = HospitalObservation
State = HospitalState

__all__ = [
    "HospitalAction",
    "HospitalObservation",
    "HospitalState",
    "Action",
    "Observation",
    "State",
]