diff --git a/nifty6/minimization/metric_gaussian_kl.py b/nifty6/minimization/metric_gaussian_kl.py index 85122e2817093ca8cbc642e19510cd275c848765..bfde7af5bd336aa26c0e63fbd0f7b256850cb978 100644 --- a/nifty6/minimization/metric_gaussian_kl.py +++ b/nifty6/minimization/metric_gaussian_kl.py @@ -156,18 +156,18 @@ class MetricGaussianKL(Energy): self._hamiltonian = hamiltonian + self._n_samples = int(n_samples) self._use_mpi = bool(use_mpi) if self._use_mpi: from mpi4py import MPI self._comm = MPI.COMM_WORLD - self._ntask = self._comm.Get_size() - self._rank = self._comm.Get_rank() - self._master = (self._rank == 0) + ntask = self._comm.Get_size() + rank = self._comm.Get_rank() + self._lo, self._hi = _shareRange(self._n_samples, ntask, rank) else: - self._comm, self._ntask, self._rank, self._master = None, 1, 0, True + self._comm = None + self._lo, self._hi = 0, self._n_samples - self._n_samples = int(n_samples) - self._lo, self._hi = _shareRange(self._n_samples, self._ntask, self._rank) self._mirror_samples = bool(mirror_samples) self._n_eff_samples = self._n_samples if self._mirror_samples: