Change prng key.

astro-informatics · Aug 14, 2024 · 0c7e824 · 0c7e824
1 parent b1560e1
commit 0c7e824
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_flow_model.py b/tests/test_flow_model.py
@@ -245,7 +245,7 @@ def test_flows_gaussian(model):
         epochs = 20
 
     # Initialize a PRNG key (you can use any valid key)
-    key = jax.random.PRNGKey(0)
+    key = jax.random.PRNGKey(2)
     mean = jnp.zeros(ndim)
     cov = jnp.eye(ndim)