explictly include jax=True

lnccbrown · May 7, 2024 · 3391ae4 · 3391ae4
1 parent e586ec0
commit 3391ae4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/slow/test_mcmc.py b/tests/slow/test_mcmc.py
@@ -10,7 +10,7 @@
 
 from hssm.utils import _rearrange_data
 
-hssm.set_floatX("float32")
+hssm.set_floatX("float32", jax=True)
 
 # AF-TODO: Include more tests that use different link functions!