Skip to content

Commit

Permalink
Convert the rest of the type conflicts
Browse files Browse the repository at this point in the history
  • Loading branch information
kazewong committed Jul 23, 2024
1 parent 0080e28 commit 602c164
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions src/jimgw/single_event/utils.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
import jax.numpy as jnp
from jax import jit
from jax.scipy.integrate import trapezoid
from jax.scipy.special import i0e
from jaxtyping import Array, Float
Expand Down Expand Up @@ -140,7 +139,7 @@ def ra_dec_to_theta_phi(ra: Float, dec: Float, gmst: Float) -> tuple[Float, Floa
return theta, phi


def euler_rotation(delta_x: tuple[Float, Float, Float]):
def euler_rotation(delta_x: Float[Array, " 3"]):
"""
Calculate the rotation matrix mapping the vector (0, 0, 1) to delta_x
while preserving the origin of the azimuthal angle.
Expand Down Expand Up @@ -260,7 +259,7 @@ def theta_phi_to_ra_dec(theta: Float, phi: Float, gmst: Float) -> tuple[Float, F


def zenith_azimuth_to_ra_dec(
zenith: Float, azimuth: Float, gmst: Float, delta_x: tuple[Float, Float, Float]
zenith: Float, azimuth: Float, gmst: Float, delta_x: Float[Array, " 3"]
) -> tuple[Float, Float]:
"""
Transforming the azimuthal angle and zenith angle in Earth frame to right ascension and declination.
Expand Down

0 comments on commit 602c164

Please sign in to comment.