Fix untested code

blackjax-devs · Dec 2, 2023 · 149ec31 · 149ec31
1 parent 18ee90c
commit 149ec31
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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__":