class BaseAgent(ABC):
def reset(self, *args: Any, **kwargs: Any):
def step(self, *args: Any, **kwargs: Any):