Changing the way of importing SkyGrid class and correcting the import

of pycbc.conversions
gwastro · Jan 9, 2025 · b9367d4 · b9367d4
1 parent 8da4055
commit b9367d4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/bin/pycbc_make_sky_grid b/bin/pycbc_make_sky_grid
@@ -19,7 +19,7 @@ from scipy.spatial.transform import Rotation as R
 import pycbc
 from pycbc.detector import Detector
 from pycbc.types import angle_as_radians
-from pycbc.tmpltbank import sky_grid
+from pycbc.tmpltbank.sky_grid import SkyGrid
 
 
 def spher_to_cart(sky_points):
@@ -164,7 +164,7 @@ rota = r.apply(cart)
 # Convert cartesian coordinates back to spherical coordinates
 spher = cart_to_spher(rota)
 
-sky_grid = sky_grid.SkyGrid(
+sky_grid = SkyGrid(
     spher[:, 0], spher[:, 1], args.instruments, args.trigger_time
 )
 

diff --git a/pycbc/tmpltbank/sky_grid.py b/pycbc/tmpltbank/sky_grid.py
@@ -8,7 +8,8 @@
 import h5py
 
 from pycbc.detector import Detector
-from pycbc.conversion import ensurearray
+from pycbc.conversions import ensurearray
+
 
 class SkyGrid:
     def __init__(self, ra, dec, detectors, ref_gps_time):