fix: inpainted_regions handling

HERA-Team · May 16, 2024 · 8ed8a60 · 8ed8a60
1 parent de44a28
commit 8ed8a60
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/hera_notebook_templates/notebooks/lststack.ipynb b/hera_notebook_templates/notebooks/lststack.ipynb
@@ -2434,10 +2434,7 @@
     "    \n",
     "    bad_fellas = {}\n",
     "    \n",
-    "    nights = [\n",
-    "        zscores[0].nights.tolist(),\n",
-    "        zscores[1].nights.tolist()\n",
-    "    ]\n",
+    "    nights = [zsq.nights.tolist() for zsq in zscores]\n",
     "\n",
     "    chsizes = [(2, 5), (5, 10), (10, 20)]    \n",
     "    sized = {ch: {} for ch in chsizes}\n",
@@ -2456,7 +2453,7 @@
     "        keys = list(bads.keys())\n",
     "        \n",
     "        meanz = np.array([\n",
-    "            np.nanmean(zscores[lst].metrics[nights.index(jdint), zscores[lst].antpairs.index((a,b)), low:high, pol]) \n",
+    "            np.nanmean(zscores[lst].metrics[nights[lst].index(jdint), zscores[lst].antpairs.index((a,b)), low:high, pol]) \n",
     "            for (lst, a, b, pol, jdint, low, high) in bads.keys()\n",
     "        ])\n",
     "        \n",