hanabhi/gridworld-env

1from typing import Tuple2from core.env_server import Action, Observation, State3 4class GridAction(Action):5    direction: str6 7class GridObservation(Observation):8    agent_pos: Tuple[int, int]9    goal_pos: Tuple[int, int]10    reward: float11    done: bool12 13class GridState(State):14    episode_id: str15    steps_taken: int16    grid_size: int = 5