diff --git a/nifty6/minimization/metric_gaussian_kl.py b/nifty6/minimization/metric_gaussian_kl.py index 8cfe2a4e5f8dc0d41c666eafdd29c17c73878719..c9ef5b0f73582458a5e20143594e19403ede994c 100644 --- a/nifty6/minimization/metric_gaussian_kl.py +++ b/nifty6/minimization/metric_gaussian_kl.py @@ -257,7 +257,7 @@ class MetricGaussianKL(Energy): res = res + tuple(-item for item in res) return res - def _unscaled_metric_sample(self, from_inverse=False, dtype=np.float64): + def _metric_sample(self, from_inverse=False, dtype=np.float64): if from_inverse: raise NotImplementedError() lin = self._lin.with_want_metric() @@ -269,7 +269,4 @@ class MetricGaussianKL(Energy): if self._mirror_samples: samp = samp + self._hamiltonian(lin-v).metric.draw_sample(from_inverse=False, dtype=dtype) random.pop_sseq() - return _allreduce_sum_field(self._comm, samp) - - def _metric_sample(self, from_inverse=False, dtype=np.float64): - return self._unscaled_metric_sample(from_inverse, dtype)/self._n_eff_samples + return _allreduce_sum_field(self._comm, samp)/self._n_eff_samples