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