figure out proper time slice based on flags in both polarizations

HERA-Team · Jun 19, 2024 · fc3ecf7 · fc3ecf7
1 parent 019df0c
commit fc3ecf7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/notebooks/single_baseline_postprocessing_and_pspec.ipynb b/notebooks/single_baseline_postprocessing_and_pspec.ipynb
@@ -451,9 +451,9 @@
     "print(f'Below FM Frequency Slice: {low_band}')\n",
     "print(f'Above FM Frequency Slice: {high_band}')\n",
     "\n",
-    "# figure out the range of unflagged times\n",
-    "tslice = slice(true_stretches(~np.all(flags[bl], axis=1))[0].start, true_stretches(~np.all(flags[bl], axis=1))[-1].stop)\n",
-    "\n",
+    "# figure out the range of unflagged times for both polarizations\n",
+    "ORed_flags = np.all(flags[ANTPAIR + ('ee',)], axis=1) | np.all(flags[ANTPAIR + ('nn',)], axis=1)\n",
+    "tslice = slice(true_stretches(~ORed_flags)[0].start, true_stretches(~ORed_flags)[-1].stop)\n",
     "print(f'Time Slice Excluded Edge Flags: {tslice}')"
    ]
   },