diff --git a/nifty5/minimization/kl_energy.py b/nifty5/minimization/kl_energy.py index 3b12f469816db1dce987ee4a40ab5e8530d741d2..0e8155f9e3d4bd36c40433bdb94a61459117010e 100644 --- a/nifty5/minimization/kl_energy.py +++ b/nifty5/minimization/kl_energy.py @@ -8,7 +8,7 @@ from .. import utilities class KL_Energy(Energy): def __init__(self, position, h, nsamp, constants=[], - constants_samples=None, mirror_samples=False, + constants_samples=None, gen_mirrored_samples=False, _samples=None): super(KL_Energy, self).__init__(position) if h.domain is not position.domain: @@ -18,13 +18,12 @@ class KL_Energy(Energy): if constants_samples is None: constants_samples = constants self._constants_samples = constants_samples - self._mirror_samples = mirror_samples if _samples is None: met = h(Linearization.make_partial_var( position, constants_samples, True)).metric _samples = tuple(met.draw_sample(from_inverse=True) for _ in range(nsamp)) - if mirror_samples: + if gen_mirrored_samples: _samples += tuple(-s for s in _samples) self._samples = _samples @@ -43,9 +42,9 @@ class KL_Energy(Energy): self._metric = None def at(self, position): - return KL_Energy(position, self._h, 0, self._constants, - self._constants_samples, self._mirror_samples, - self._samples) + return KL_Energy(position, self._h, 0, + self._constants, self._constants_samples, + _samples = self._samples) @property def value(self):