From cdbdc0ef0b92057668e7314c6a11cadd9acb2939 Mon Sep 17 00:00:00 2001
From: ahmedk <karim.ahmed@xfel.eu>
Date: Wed, 31 Jan 2024 15:42:55 +0100
Subject: [PATCH] Correcting one train for epix100 and storing a list of one
pulseId
---
notebooks/ePix100/Correction_ePix100_NBC.ipynb | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/notebooks/ePix100/Correction_ePix100_NBC.ipynb b/notebooks/ePix100/Correction_ePix100_NBC.ipynb
index b7db67227..7bc7b208d 100644
--- a/notebooks/ePix100/Correction_ePix100_NBC.ipynb
+++ b/notebooks/ePix100/Correction_ePix100_NBC.ipynb
@@ -635,8 +635,11 @@
" \"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",
" outp_source.create_key(\n",
- " \"data.pulseId\", data=list(seq_dc[instrument_src]['data.pulseId'].ndarray().squeeze()), chunks=min(50, len(seq_dc.train_ids)))\n",
+ " \"data.pulseId\", data=pulse_ids, 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