From eb1841a8992436a54fcedfd503592ae120ac8bac Mon Sep 17 00:00:00 2001
From: Cammille Carinan <cammille.carinan@xfel.eu>
Date: Tue, 22 Feb 2022 09:17:00 +0100
Subject: [PATCH] Improve SCS_slowTrain call

---
 src/toolbox_scs/detectors/hrixs.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/toolbox_scs/detectors/hrixs.py b/src/toolbox_scs/detectors/hrixs.py
index 0f6ead7..b3a8cb4 100644
--- a/src/toolbox_scs/detectors/hrixs.py
+++ b/src/toolbox_scs/detectors/hrixs.py
@@ -307,12 +307,10 @@ class hRIXS:
         if proposal is None:
             proposal = cls.PROPOSAL
 
-        run, data = tb.load(proposal,
-                            runNB=runNB,
-                            fields=['hRIXS_det', 'SCS_slowTrain'])
+        run, data = tb.load(proposal, runNB=runNB, fields=['hRIXS_det'])
 
         # Get slow train data
-        mnemo = tb.mnemonics['SCS_slowTrain'][0]
+        mnemo = tb.mnemonics_for_run(run)['SCS_slowTrain']
         slow_train = run[mnemo['source'], mnemo['key']].ndarray().sum()
 
         return cls(images=data['hRIXS_det'][1 if first_wrong else 0:].data,
-- 
GitLab