diff --git a/python/bartrs/compile_pymc.py b/python/bartrs/compile_pymc.py
index f94246a..d8048cf 100644
--- a/python/bartrs/compile_pymc.py
+++ b/python/bartrs/compile_pymc.py
@@ -232,7 +232,6 @@ def _make_persistent_arrays(self):
         the compiled function interface.
         """
         arrays = [item.storage[0].copy() for item in self.logp_fn_ptr.input_storage[1:]]
-        assert all(arr.dtype == np.float64 for arr in arrays)
         return arrays
 
     # TODO: fast update for shared arrays