From 3391ae484723322f7b2c026af7e8d2e660e625ab Mon Sep 17 00:00:00 2001
From: Fengler <alexanderfengler@gmx.de>
Date: Tue, 7 May 2024 22:28:19 +0200
Subject: [PATCH] explictly include jax=True

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

diff --git a/tests/slow/test_mcmc.py b/tests/slow/test_mcmc.py
index 96f86a0d..83b24347 100644
--- 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!