enforce that vis_units are Jy

HERA-Team · Jul 11, 2024 · 16fc632 · 16fc632
1 parent b4687fd
commit 16fc632
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/notebooks/file_calibration.ipynb b/notebooks/file_calibration.ipynb
@@ -1693,7 +1693,9 @@
     "                         flags={bl_to_red_map[bl]: vissol_flags[bl] for bl in vissol_flags}, \n",
     "                         nsamples={bl_to_red_map[bl]: vissol_nsamples[bl] for bl in vissol_nsamples})\n",
     "        hd_vissol.pol_convention = hd_auto_model.pol_convention\n",
-    "        hd_vissol.vis_units = hd_auto_model.vis_units\n",
+    "        if hd_auto_model.vis_units != 'Jy':\n",
+    "            raise ValueError(f'The HERA pipeline is hard-coded for calibrated visibilities in Jy, but somehow this abscal model is in {hd_auto_model.vis_units}')\n",
+    "        hd_vissol.vis_units = 'Jy'\n",
     "        hd_vissol.write_uvh5(OMNIVIS_FILE, clobber=True)"
    ]
   },