From 0c7e824f3ff5d9932340999942903b986da9df11 Mon Sep 17 00:00:00 2001
From: alicjapolanska <alicja.polanska.22@ucl.ac.uk>
Date: Wed, 14 Aug 2024 14:16:37 +0100
Subject: [PATCH] Change prng key.

---
 tests/test_flow_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_flow_model.py b/tests/test_flow_model.py
index ea00bb6..3964415 100644
--- 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)