From c40da307b8a3c560c24d6b2da1c918c285ed127e Mon Sep 17 00:00:00 2001
From: Philipp Arras <parras@mpa-garching.mpg.de>
Date: Thu, 15 Jul 2021 12:07:01 +0200
Subject: [PATCH] KL energies: actually use napprox
---
src/minimization/kl_energies.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/minimization/kl_energies.py b/src/minimization/kl_energies.py
index 2a9c08e3c..caf31d0e1 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,
--
GitLab