dataclass class AgentState: i: int action: int pos = None collision_vector = None action_valid = None