From bc159d5f62eabd7ec9843d29f4228a5139f9922a Mon Sep 17 00:00:00 2001
From: ahmedk <karim.ahmed@xfel.eu>
Date: Thu, 23 Mar 2023 10:41:14 +0100
Subject: [PATCH] error out only if all sources are empty
---
.../AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb | 10 +++++++---
1 file changed, 7 insertions(+), 3 deletions(-)
diff --git a/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb b/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
index bbb32e928..f05b3cf1b 100644
--- a/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
+++ b/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
@@ -176,12 +176,16 @@
"outputs": [],
"source": [
"run_dc = RunDirectory(in_folder / f\"r{run:04d}\")\n",
- "\n",
"instrument_src = instrument_source_template.format(karabo_id, receiver_template)\n",
"\n",
- "instr_dc = run_dc.select(instrument_src.format(\"*\"), require_all=True)\n",
+ "instr_dc = run_dc.select(instrument_src.format(\"*\"))\n",
+ "\n",
+ "for m in modules:\n",
+ " # Remove empty sources from `instr_dc`\n",
+ " if instr_dc[instrument_src.format(m), 'image.data'].shape[0] == 0:\n",
+ " instr_dc = instr_dc.deselect(instrument_src.format(m))\n",
"\n",
- "if not instr_dc.train_ids:\n",
+ "if not instr_dc.all_sources:\n",
" raise ValueError(f\"No images found for {in_folder / f'r{run:04d}'}\")"
]
},
--
GitLab