ce96b03b07a4a4bdd851aa84493c616cd291aff2,rllib/agents/mbmpo/model_ensemble.py,DynamicsEnsembleCustomModel,predict_model_batches,#DynamicsEnsembleCustomModel#Any#Any#Any#,323

Before Change

        delta = self.forward(x).detach().numpy()
        if self.normalize_data:
            delta = denormalize(delta, self.normalizations["delta"])
        return pre_obs + delta

    def set_norms(self, normalization_dict):
        self.normalizations = normalization_dict

After Change

                                self.normalizations[SampleBatch.ACTIONS])
        x = np.concatenate([obs, actions], axis=-1)
        x = convert_to_torch_tensor(x, device=device)
        delta = self.forward(x).detach().cpu().numpy()
        if self.normalize_data:
            delta = denormalize(delta, self.normalizations["delta"])
        new_obs = pre_obs + delta
        clipped_obs = np.clip(new_obs, self.env_obs_space.low,
                              self.env_obs_space.high)
        return clipped_obs

    def set_norms(self, normalization_dict):
        self.normalizations = normalization_dict

Instances