Force JAX double precision mode in Wigner precompute tests

Author: matt-graham

tests/test_wigner_precompute.py

@@ -1,3 +1,4 @@
+import jax
 import numpy as np
 import pytest
 import so3
@@ -8,6 +9,8 @@
 from s2fft.precompute_transforms.wigner import forward, inverse
 from s2fft.sampling import so3_samples as samples
 
+jax.config.update("jax_enable_x64", True)
+
 L_to_test = [6]
 N_to_test = [2, 6]
 nside_to_test = [4]