From 149ec310a7b15b79195d5f2eb67e73c6814743d8 Mon Sep 17 00:00:00 2001 From: Adrien Corenflos Date: Sat, 2 Dec 2023 17:32:07 +0200 Subject: [PATCH] Fix untested code --- tests/vi/test_schrodinger_follmer.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/vi/test_schrodinger_follmer.py b/tests/vi/test_schrodinger_follmer.py index c1fe14f13..db5451e49 100644 --- a/tests/vi/test_schrodinger_follmer.py +++ b/tests/vi/test_schrodinger_follmer.py @@ -78,9 +78,7 @@ def logp_unnormalized_posterior(x, observed, prior_mu, prior_prec, true_cov): ) # make sure it runs with 1 sample only - _ = schrodinger_follmer_algo.sample( - rng_key_init, initial_state, 1 - ) + _ = schrodinger_follmer_algo.sample(rng_key_init, initial_state, 1) if __name__ == "__main__":