don't save pickle of save results is false

HERA-Team · Jul 16, 2024 · ca75979 · ca75979
1 parent eae90a1
commit ca75979
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/notebooks/file_postprocessing.ipynb b/notebooks/file_postprocessing.ipynb
@@ -325,8 +325,9 @@
     "flagged_bls = [bl for bl in flags if np.all(flags[bl])]\n",
     "flagged_ants = [ant for ant in cal_flags if np.all(cal_flags[ant])]\n",
     "reds_used_here = redcal.filter_reds(reds, ex_bls=flagged_bls, ex_ants=flagged_ants)\n",
-    "with open(REDS_PICKLE_FILE, 'wb') as reds_pickle_file:\n",
-    "    pickle.dump(reds_used_here, reds_pickle_file)"
+    "if SAVE_RESULTS:\n",
+    "    with open(REDS_PICKLE_FILE, 'wb') as reds_pickle_file:\n",
+    "        pickle.dump(reds_used_here, reds_pickle_file)"
    ]
   },
   {