def METHOD_NAME(self, **kwargs):
    """Resets the environment and normalizes the observation."""
    obs, info = self.env.METHOD_NAME(**kwargs)
    if self.is_vector_env:
        return self.normalize(obs), info
    else:
        return self.normalize(np.array([obs]))[0], info