From 67cef9ef77b795396b26e63c0f230fb1feb9a452 Mon Sep 17 00:00:00 2001
From: Martin Teichmann <martin.teichmann@xfel.eu>
Date: Thu, 8 Dec 2022 12:51:29 +0100
Subject: [PATCH] modify dataset instead of returning a copy
---
src/toolbox_scs/detectors/hrixs.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/src/toolbox_scs/detectors/hrixs.py b/src/toolbox_scs/detectors/hrixs.py
index 0c7a329..e1dd0ac 100644
--- a/src/toolbox_scs/detectors/hrixs.py
+++ b/src/toolbox_scs/detectors/hrixs.py
@@ -495,10 +495,11 @@ class hRIXS:
rvy, _ = np.histogram(
ysfp1.ravel(), weights=right.ravel(), **kwargs)
r[:] = (lhy + rhy) / (lvy + rvy)
- data = data.assign_coords(
- energy=np.arange(self.Y_RANGE.start + 10, self.Y_RANGE.stop - 10)
- * self.ENERGY_SLOPE + self.ENERGY_INTERCEPT)
- return data.assign(spectrum=(("trainId", "energy"), ret))
+ data.coords["energy"] = \
+ np.arange(self.Y_RANGE.start + 10, self.Y_RANGE.stop - 10) \
+ * self.ENERGY_SLOPE + self.ENERGY_INTERCEPT
+ data['spectrum'] = (("trainId", "energy"), ret)
+ return data
aggregators = dict(
hRIXS_det=lambda x, dim: x.sum(dim=dim),
--
GitLab