handle outriggerless data

HERA-Team · Dec 26, 2024 · a89f27e · a89f27e
1 parent df60e89
commit a89f27e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/notebooks/antenna_classification_summary.ipynb b/notebooks/antenna_classification_summary.ipynb
@@ -600,9 +600,11 @@
    "source": [
     "def array_class_plot():\n",
     "    fig, axes = plt.subplots(1, 2, figsize=(14, 6), dpi=100, gridspec_kw={'width_ratios': [2, 1]})\n",
-    "    plot_antclass(hd.antpos, overall_class, ax=axes[0], ants=[ant for ant in hd.data_ants if ant < 320], legend=False, \n",
-    "                  title=f'HERA Core: Overall Flagging Based on {overall_thresh:.1%} Daily Threshold')\n",
-    "    plot_antclass(hd.antpos, overall_class, ax=axes[1], ants=[ant for ant in hd.data_ants if ant >= 320], radius=50, title='Outriggers')"
+    "    if len([ant for ant in hd.data_ants if ant < 320]) > 0:\n",
+    "        plot_antclass(hd.antpos, overall_class, ax=axes[0], ants=[ant for ant in hd.data_ants if ant < 320], legend=False, \n",
+    "                      title=f'HERA Core: Overall Flagging Based on {overall_thresh:.1%} Daily Threshold')\n",
+    "    if len([ant for ant in hd.data_ants if ant >= 320]) > 0:\n",
+    "        plot_antclass(hd.antpos, overall_class, ax=axes[1], ants=[ant for ant in hd.data_ants if ant >= 320], radius=50, title='Outriggers')"
    ]
   },
   {