add check for missing files in visualization notebook (#307)

Co-authored-by: Stefanos Chatzimichelakis <s.chatzimichelakis@iccs.gr>
geoschem · Jan 25, 2025 · 45b6ce5 · 45b6ce5
1 parent bc609ab
commit 45b6ce5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/notebooks/visualization_notebook.ipynb b/src/notebooks/visualization_notebook.ipynb
@@ -584,6 +584,9 @@
     "        pth_posterior = os.path.join(data_posterior, f)\n",
     "        # Load TROPOMI/GEOS-Chem and Jacobian matrix data from the .pkl file\n",
     "        obj = load_obj(pth)\n",
+    "        # If there is no corresponding file in data_posterior dir, skip\n",
+    "        if not os.path.isfile(pth_posterior):\n",
+    "            continue\n",
     "        obj_posterior = load_obj(pth_posterior)\n",
     "        # If there aren't any TROPOMI observations on this day, skip\n",
     "        if obj[\"obs_GC\"].shape[0] == 0:\n",