diff --git a/src/minimization/kl_energies.py b/src/minimization/kl_energies.py index 2a9c08e3c48deb2bd294c86c2618db126510948a..caf31d0e1e5590aea105558680b6e62cbc6298b8 100644 --- a/src/minimization/kl_energies.py +++ b/src/minimization/kl_energies.py @@ -407,7 +407,7 @@ def MetricGaussianKL(mean, hamiltonian, n_samples, mirror_samples, constants=[], _, ham_sampling = _reduce_by_keys(mean, hamiltonian, point_estimates) sampler = _MetricGaussianSampler(mean, ham_sampling, n_samples, - mirror_samples) + mirror_samples, napprox) local_samples = sampler.draw_samples(comm) mean, hamiltonian = _reduce_by_keys(mean, hamiltonian, constants) @@ -517,7 +517,8 @@ def GeoMetricKL(mean, hamiltonian, n_samples, minimizer_samp, mirror_samples, _, ham_sampling = _reduce_by_keys(mean, hamiltonian, point_estimates) sampler = _GeoMetricSampler(mean, ham_sampling, minimizer_samp, - start_from_lin, n_samples, mirror_samples) + start_from_lin, n_samples, mirror_samples, + napprox) local_samples = sampler.draw_samples(comm) mean, hamiltonian = _reduce_by_keys(mean, hamiltonian, constants) return _SampledKLEnergy(mean, hamiltonian, sampler.n_eff_samples, False,