make sure code is covered. There's no logic tested.

blackjax-devs · Dec 2, 2023 · e459136 · e459136
1 parent 149ec31
commit e459136
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/vi/test_schrodinger_follmer.py b/tests/vi/test_schrodinger_follmer.py
@@ -79,6 +79,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.step(rng_key_init, initial_state)
 
 
 if __name__ == "__main__":