fix pre-commit

Author: ASKabalan

tests/test_healpix_ffts.py

@@ -112,9 +112,6 @@ def healpix_jax(f):
     def healpix_cuda(f):
         return healpix_fft_cuda(f, L, nside, False).real
 
-    vmapped_jax = jax.vmap(healpix_jax)(f_stacked)
-    vmapped_cuda = jax.vmap(healpix_cuda)(f_stacked)
-
     f = f_stacked[0]
     # Test VMAP
     MSE = jnp.mean(