wip

tsunhopang · Aug 13, 2024 · 0a2e68c · 0a2e68c
1 parent d5f86e5
commit 0a2e68c
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/jimgw/single_event/transforms.py b/src/jimgw/single_event/transforms.py
@@ -208,7 +208,7 @@ def __init__(
 
         def named_transform(x):
             t_det = x["t_c"] + self.ifo.delay_from_geocenter(
-                x["ra"], x["dec"], self.gmst
+                x["ra"][0], x["dec"][0], self.gmst
             )
             return {
                 "t_det": t_det,
@@ -217,8 +217,9 @@ def named_transform(x):
         self.transform_func = named_transform
 
         def named_inverse_transform(x):
+            import pdb; pdb.set_trace()
             t_c = x["t_det"] - self.ifo.delay_from_geocenter(
-                x["ra"], x["dec"], self.gmst
+                x["ra"][0], x["dec"][0], self.gmst
             )
             return {
                 "t_c": t_c,
@@ -268,7 +269,7 @@ def _calc_R_det(x):
             p_iota_term = (1.0 + jnp.cos(iota) ** 2) / 2.0
             c_iota_term = jnp.cos(iota)
 
-            antenna_pattern = self.ifo.antenna_pattern(ra, dec, psi, self.gmst)
+            antenna_pattern = self.ifo.antenna_pattern(ra[0], dec[0], psi[0], self.gmst)
             p_mode_term = p_iota_term * antenna_pattern["p"]
             c_mode_term = c_iota_term * antenna_pattern["c"]
 
@@ -278,7 +279,7 @@ def named_transform(x):
             R_det = _calc_R_det(x)
             phase_det = jnp.angle(R_det) + x["phase_c"] / 2.0
             return {
-                "phase_det": phase_det % (2. * jnp.pi),
+                "phase_det": phase_det % (2.0 * jnp.pi),
             }
 
         self.transform_func = named_transform
@@ -287,7 +288,7 @@ def named_inverse_transform(x):
             R_det = _calc_R_det(x)
             phase_c = (-jnp.angle(R_det) + x["phase_det"]) * 2.0
             return {
-                "phase_c": phase_c % (2. * jnp.pi),
+                "phase_c": phase_c % (2.0 * jnp.pi),
             }
 
         self.inverse_transform_func = named_inverse_transform