From 81338215e9e95675fe3f7d83f2a7b03ea2abb0be Mon Sep 17 00:00:00 2001
From: Laurent Mercadier <laurent.mercadier@xfel.eu>
Date: Thu, 27 Feb 2025 09:39:29 +0100
Subject: [PATCH] Convert index back to integer values after where() operation
---
src/toolbox_scs/detectors/viking.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/toolbox_scs/detectors/viking.py b/src/toolbox_scs/detectors/viking.py
index 476c527..ff96db9 100644
--- a/src/toolbox_scs/detectors/viking.py
+++ b/src/toolbox_scs/detectors/viking.py
@@ -290,6 +290,7 @@ class Viking:
"""
if key not in data:
return
+ print('hello')
x = data.newt_x
spectra = data[key]
mask = xr.ones_like(x, dtype=bool)
@@ -300,7 +301,8 @@ class Viking:
ranges = self.BL_SIGNAL_RANGE
for xrange in ranges:
mask = mask & ((x < xrange[0]) | (x > xrange[1]))
- x_bl = x.where(mask, drop=True)
+ x_bl = x.where(mask, drop=True).astype(int)
+ print(x_bl)
bl = spectra.sel(newt_x=x_bl)
fit = np.polyfit(x_bl, bl.T, self.BL_POLY_DEG)
if len(spectra.shape) == 1:
--
GitLab