Remove default

patrick-kidger · Nov 7, 2023 · afcefd7 · afcefd7
1 parent d803c1e
commit afcefd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/helpers.py b/tests/helpers.py
@@ -33,7 +33,7 @@ def getkey():
 
 
 @ft.lru_cache(maxsize=None)
-def _construct_matrix_impl(getkey, cond_cutoff, tags, size, dtype=jnp.float64):
+def _construct_matrix_impl(getkey, cond_cutoff, tags, size, dtype):
     while True:
         matrix = jr.normal(getkey(), (size, size), dtype=dtype)
         if has_tag(tags, lx.diagonal_tag):