bug fix for jnp.complex128 in Gtilde in tf

Hazboun6 · Dec 31, 2024 · 881f392 · 881f392
1 parent c8650c6
commit 881f392
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/hasasia/sensitivity.py b/hasasia/sensitivity.py
@@ -169,9 +169,9 @@ def get_Tf(designmatrix, toas, N=None, nf=200, fmin=None, fmax=2e-7,
         else:
             G = Gmatrix
         m = G.shape[1]
-        Gtilde = np.zeros((ff.size,G.shape[1]),dtype='complex128')
-        Gtilde = np.dot(np.exp(1j*2*np.pi*ff[:,np.newaxis]*toas),G)
-        Tmat = jnp.matmul(np.conjugate(Gtilde),Gtilde.T)/N_TOA
+        Gtilde = jnp.zeros((ff.size, G.shape[1]), dtype=jnp.complex128)
+        Gtilde = np.dot(np.exp(1j*2*np.pi*ff[:,np.newaxis]*toas),G).astype(jnp.complex128)
+        Tmat = jnp.matmul(jnp.conjugate(Gtilde),Gtilde.T)/N_TOA
         if twofreqs:
             Tmat = np.real(Tmat)
         else: