Merge branch 'default_qutrit_mixed_execute' of github.com:QSAR-UBC/pe…

…nnylane_qutrit_mixer into default_qutrit_mixed_execute
QSAR-UBC · Apr 20, 2024 · f535b0b · f535b0b
2 parents d3cbcf6 + f48b15b
commit f535b0b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/devices/test_default_qutrit_mixed.py b/tests/devices/test_default_qutrit_mixed.py
@@ -1058,6 +1058,9 @@ class TestPRNGKeySeed:
 
     def test_prng_key_as_seed(self):
         """Test that a jax PRNG can be passed as a seed."""
+        from jax.config import config
+        config.update("jax_enable_x64", True)
+
         from jax import random
 
         key1 = random.key(123)