def postprocess_observations(self, observation_batch: th.Tensor):
    return ((observation_batch + 1) / 2)
def postprocess_observations(self, observation_batch: th.Tensor):
    return ((observation_batch + 1) / 2)