From 25b040c85dc011b94b9847ddaf14bacca8a0283a Mon Sep 17 00:00:00 2001
From: ahmedk <karim.ahmed@xfel.eu>
Date: Tue, 15 Mar 2022 11:20:24 +0100
Subject: [PATCH] fix plots based on selected cellIds
---
notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index 9c38bd80c..679a3172d 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -882,7 +882,8 @@
" \"\"\"\n",
" run_data = RunDirectory(data_folder, include)\n",
" if tid is not None:\n",
- " tid, data = run_data.select(f'{detector_id}/DET/*', source).train_from_id(tid)\n",
+ " tid, data = run_data.select(\n",
+ " f'{detector_id}/DET/*', source).train_from_id(tid, keep_dims=True)\n",
" else:\n",
" tid, data = next(iter(run_data.select(f'{detector_id}/DET/*', source).trains(require_all=True)))\n",
" \n",
--
GitLab