From 03ff42d99d4df079a8162fe161eeecc0e2073cf0 Mon Sep 17 00:00:00 2001
From: ahmedk <karim.ahmed@xfel.eu>
Date: Thu, 1 Feb 2024 17:15:39 +0100
Subject: [PATCH] preserve pulseid dtype
---
notebooks/ePix100/Correction_ePix100_NBC.ipynb | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/notebooks/ePix100/Correction_ePix100_NBC.ipynb b/notebooks/ePix100/Correction_ePix100_NBC.ipynb
index 7bc7b208d..7df412a55 100644
--- a/notebooks/ePix100/Correction_ePix100_NBC.ipynb
+++ b/notebooks/ePix100/Correction_ePix100_NBC.ipynb
@@ -635,11 +635,12 @@
" \"data.trainId\", data=seq_dc.train_ids, chunks=min(50, len(seq_dc.train_ids)))\n",
" \n",
" if np.isin('data.pulseId', list(seq_dc[instrument_src].keys())): # some runs are missing 'data.pulseId'\n",
- " pulse_ids = seq_dc[instrument_src]['data.pulseId'].ndarray().squeeze().tolist()\n",
- " if isinstance(pulse_ids, int): # one pulse in case of correcting one train\n",
- " pulse_ids = [pulse_ids]\n",
+ " pulseid = seq_dc[instrument_src]['data.pulseId'].ndarray().squeeze()\n",
+ " # pulseid is scalar for one train.\n",
+ " pulseid = [pulseid] if not pulseid.shape else list(pulseid)\n",
+ "\n",
" outp_source.create_key(\n",
- " \"data.pulseId\", data=pulse_ids, chunks=min(50, len(seq_dc.train_ids)))\n",
+ " \"data.pulseId\", data=pulseid, chunks=min(50, len(seq_dc.train_ids)))\n",
" \n",
" if pattern_classification:\n",
" # Add main corrected `data.image.pixels` dataset and store corrected data.\n",
--
GitLab