if self.hps.co_dim > 0:
        controller_outputs[es_idx,:,:] = model_values['controller_outputs']