Patch for importing VLA uvfits files

Patch for error in importing VLA uvfits files exported from CASA
achael · Nov 16, 2020 · 65d94a9 · 65d94a9
1 parent 50e728c
commit 65d94a9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ehtim/io/load.py b/ehtim/io/load.py
@@ -1307,8 +1307,8 @@ def load_obs_uvfits(filename, polrep='stokes', flipbl=False, allow_singlepol=Tru
     t2 = data['BASELINE'][mask].astype(int) - t1 * 256
     t1 = t1 - 1
     t2 = t2 - 1
-    t1 = np.array([tarr[i]['site'] for i in t1])
-    t2 = np.array([tarr[i]['site'] for i in t2])
+    t1 = np.array([tarr[np.where(tnums==i)[0][0]]['site'] for i in t1])
+    t2 = np.array([tarr[np.where(tnums==i)[0][0]]['site'] for i in t2])
 
     # Opacities (not in standard files)
     try: