Merge pull request #128 from wheeyeon/master

Patch for importing VLA uvfits files
achael · Dec 9, 2020 · 952a640 · 952a640
2 parents 50e728c + 65d94a9
commit 952a640
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: