Change to GroundBased2G

src/jimgw/prior.py

@@ -7,7 +7,7 @@
 from jaxtyping import Array, Float, Int, PRNGKeyArray, jaxtyped
 from beartype import beartype as typechecker
 from jimgw.single_event.utils import azimuth_zenith_to_ra_dec
-from jimgw.single_event.detector import Detector, detector_preset
+from jimgw.single_event.detector import GroundBased2G, detector_preset
 from astropy.time import Time
 
 
@@ -409,7 +409,7 @@ def __init__(self, naming: str, gps: Float, ifos: list, **kwargs):
             return ValueError("At least two detectors are needed to define the Earth frame")
         elif isinstance(ifos[0], str):
             self.ifos = [detector_preset[ifo] for ifo in ifos[:2]]
-        elif isinstance(ifos[0], Detector):
+        elif isinstance(ifos[0], GroundBased2G):
             self.ifos = ifos[:2]
         else:
             return ValueError("ifos should be a list of detector names or Detector objects")

src/jimgw/single_event/utils.py

@@ -3,7 +3,7 @@
 from jax.scipy.integrate import trapezoid
 from jax.scipy.special import i0e
 from jaxtyping import Array, Float
-from jimgw.single_event.detector import Detector
+from jimgw.single_event.detector import GroundBased2G
 
 
 @jit
@@ -256,7 +256,7 @@ def theta_phi_to_ra_dec(theta: Float, phi: Float, gmst: Float) -> tuple[Float, F
 
 
 @jit
-def azimuth_zenith_to_ra_dec(azimuth: Float, zenith: Float, gmst: Float, ifos: list[Detector]) -> tuple[Float, Float]:
+def azimuth_zenith_to_ra_dec(azimuth: Float, zenith: Float, gmst: Float, ifos: list[GroundBased2G]) -> tuple[Float, Float]:
     """
     Transforming the azimuthal angle and zenith angle in Earth frame to right ascension and declination.