def _update_obs_estimate(self, obs):
    if (not self.eval_mode):
        self.obs_rms.update(obs[:, :self.obs_dim])
def _update_obs_estimate(self, obs):
    if (not self.eval_mode):
        self.obs_rms.update(obs[:, :self.obs_dim])