From 9fb0014a99a9ff8356dfe976722c41dbd6912739 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sat, 21 Dec 2024 18:31:07 +0100 Subject: [PATCH] Updated tests (no check kl for w2) --- tests/test_projections.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_projections.py b/tests/test_projections.py index 222b0ea..d71ad04 100644 --- a/tests/test_projections.py +++ b/tests/test_projections.py @@ -94,8 +94,8 @@ def test_diagonal_projection(ProjectionClass, needs_cpp, gaussian_params): assert jnp.all(jnp.isfinite(proj_params["scale"])) assert jnp.all(proj_params["scale"] > 0) - # Only check KL bounds for KL projection (and W2, which should approx hold as well) - if ProjectionClass in [KLProjection, WassersteinProjection]: + # Only check KL bounds for KL projection + if ProjectionClass in [KLProjection]: kl = compute_gaussian_kl(proj_params, gaussian_params["old_params"]) max_kl = (mean_bound + cov_bound) * 1.1 # Allow 10% margin