fix one test

LSSTDESC · Jan 14, 2025 · 0366b9e · 0366b9e
1 parent 5ec2c5e
commit 0366b9e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_shear_trans.py b/tests/test_shear_trans.py
@@ -7,9 +7,9 @@
 from jax import jit, random
 from jax_galsim import GSParams
 
-from bpd.prior import (
+from bpd.sample import sample_ellip_prior
+from bpd.shear import (
     inv_shear_transformation,
-    sample_ellip_prior,
     scalar_inv_shear_transformation,
     scalar_shear_transformation,
     shear_transformation,