models.py

"""OpenEnv models at package root for CLI/schema compatibility."""

try:
    from .env.models import MiniGridAction, MiniGridObservation, MiniGridState
except ImportError:
    from env.models import MiniGridAction, MiniGridObservation, MiniGridState

__all__ = ["MiniGridAction", "MiniGridObservation", "MiniGridState"]