Change key

Author: alicjapolanska

tests/test_flow_model.py

@@ -218,7 +218,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(10)
     mean = jnp.zeros(ndim)
     cov = jnp.eye(ndim)