From 37e3e2c46b0cc844ddb5d114afae5aaf99db88f2 Mon Sep 17 00:00:00 2001
From: ahmedk <karim.ahmed@xfel.eu>
Date: Thu, 23 Feb 2023 13:31:34 +0100
Subject: [PATCH] fix gain_mode

---
 src/cal_tools/agipdlib.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/src/cal_tools/agipdlib.py b/src/cal_tools/agipdlib.py
index bacf400ba..d2a6bbed3 100644
--- a/src/cal_tools/agipdlib.py
+++ b/src/cal_tools/agipdlib.py
@@ -1416,10 +1416,9 @@ class AgipdCorrections:
             return
 
     def initialize_from_db(self, karabo_id: str, karabo_da: str,
-                           cal_db_interface: str,
-                           creation_time: datetime,
-                           memory_cells: float, bias_voltage: int,
-                           gain_setting: float, acquisition_rate: float,
+                           creation_time: datetime, memory_cells: float,
+                           bias_voltage: int, gain_setting: int,
+                           acquisition_rate: float, gain_mode: int,
                            integration_time: int, module_idx: int,
                            photon_energy: float = 9.2, client = None,
     ):
@@ -1477,7 +1476,7 @@ class AgipdCorrections:
             acquisition_rate=acquisition_rate,
             integration_time=integration_time,
             source_energy=photon_energy,
-            gain_mode=self.gain_mode,
+            gain_mode=gain_mode,
             gain_setting=gain_setting,
             event_at=creation_time,
             client=client,
@@ -1487,7 +1486,7 @@ class AgipdCorrections:
             if cname in self.gain_constants:
                 agipd_cal.gain_mode = None
             else:
-                agipd_cal.gain_mode = self.gain_mode
+                agipd_cal.gain_mode = gain_mode
             try:
                 agipd_md = agipd_cal.metadata([cname])[karabo_da]
             except CalCatError as e:
-- 
GitLab