Add missing mass matrix in missing tests.

blackjax-devs · Sep 1, 2024 · a61af36 · a61af36
1 parent d04b7a2
commit a61af36
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/mcmc/test_sampling.py b/tests/mcmc/test_sampling.py
@@ -889,7 +889,7 @@ def test_mala(self):
     @chex.all_variants(with_pmap=False)
     def test_barker(self):
         inference_algorithm = blackjax.barker_proposal(
-            self.normal_logprob, step_size=1.5
+            self.normal_logprob, step_size=1.5, inverse_mass_matrix=jnp.eye(1)
         )
         initial_state = inference_algorithm.init(jnp.array(1.0))
         self.univariate_normal_test_case(
@@ -926,7 +926,7 @@ def test_barker(self):
     },
     {
         "algorithm": blackjax.barker_proposal,
-        "parameters": {"step_size": 0.5},
+        "parameters": {"step_size": 0.5, "inverse_mass_matrix": jnp.eye(2)},
         "is_mass_matrix_diagonal": None,
     },
 ]