From 561d05431fc74cf2ece9f29e4aeac2ecd605a745 Mon Sep 17 00:00:00 2001
From: Thomas Kluyver <thomas.kluyver@xfel.eu>
Date: Wed, 21 Feb 2024 12:46:14 +0100
Subject: [PATCH] Check gain_mode keys faster

---
 src/cal_tools/agipdlib.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/cal_tools/agipdlib.py b/src/cal_tools/agipdlib.py
index c436a2b78..74c3b852b 100644
--- a/src/cal_tools/agipdlib.py
+++ b/src/cal_tools/agipdlib.py
@@ -347,9 +347,9 @@ class AgipdCtrlRuns:
         sort_values = []
         if self.gain_modes == self.adaptive_gain_modes:  # Adaptive gain # sort by patterns
             # Patterns -> DarkHG: 1, DarkMG: 2, DarkLG: 3
-            if "patternTypeIndex.value" in self.run_ctrls[0].run_dc.keys_for_source(self.ctrl_src):
+            if "patternTypeIndex" in self.run_ctrls[0].run_dc[self.ctrl_src]:
                 sort_by = "patternTypeIndex"
-            elif "expTypeIndex.value" in self.run_ctrls[0].run_dc.keys_for_source(self.ctrl_src):
+            elif "expTypeIndex" in self.run_ctrls[0].run_dc[self.ctrl_src]:
                 sort_by = "expTypeIndex"
             else:
                 raise ValueError(
-- 
GitLab