improve finding intersector baselines (for validation especially)

HERA-Team · Dec 27, 2024 · 2673a49 · 2673a49
1 parent a89f27e
commit 2673a49
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/notebooks/full_day_systematics_inspect.ipynb b/notebooks/full_day_systematics_inspect.ipynb
@@ -125,11 +125,12 @@
     "    _, _, nsamples = hd.read()\n",
     "    nsamples = RedDataContainer(nsamples, antpos=hd.antpos)\n",
     "    # try a bunch of possible intersector baselines, hoping one will work\n",
-    "    for ant2 in range(226, 233):\n",
-    "        if (np.median(nsamples[(144, ant2, 'nn')]) > 0) & (np.median(nsamples[(144, ant2, 'ee')]) > 1):\n",
-    "            bls_to_plot.append(nsamples.get_ubl_key((144, ant2, 'nn')))\n",
-    "            bls_to_plot.append(nsamples.get_ubl_key((144, ant2, 'ee')))\n",
-    "            break\n",
+    "    for ant2 in list(range(226, 233)) + list(range(207, 215)) + list(range(187, 196)):\n",
+    "        if ((144, ant2, 'ee') in nsamples) and ((144, ant2, 'nn') in nsamples):\n",
+    "            if (np.median(nsamples[(144, ant2, 'nn')]) > 0) & (np.median(nsamples[(144, ant2, 'ee')]) > 1):\n",
+    "                bls_to_plot.append(nsamples.get_ubl_key((144, ant2, 'nn')))\n",
+    "                bls_to_plot.append(nsamples.get_ubl_key((144, ant2, 'ee')))\n",
+    "                break\n",
     "    if len(bls_to_plot) >= 8:\n",
     "        break\n",
     "\n",