in understanding_rl_vision/rl_clarity/training.py [0:0]
def step_wait(self): obs, rews, dones, infos = self.venv.step_wait() rews = rews * self._scale return obs, rews, dones, infos