From 57f2a672e61ecb14c78173e52071e808d673eca6 Mon Sep 17 00:00:00 2001
From: ahmedk <karim.ahmed@xfel.eu>
Date: Wed, 1 Nov 2023 16:57:14 +0100
Subject: [PATCH] remove unneeded args and move the function

---
 ...Jungfrau_Gain_Correct_and_Verify_NBC.ipynb | 26 +++++++++----------
 1 file changed, 12 insertions(+), 14 deletions(-)

diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index afcbd7c43..95bb91ef5 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -398,6 +398,18 @@
     "    else:\n",
     "        gain_corr[index, ...] = g\n",
     "\n",
+    "    # Jungfrau gains 0[00], 1[01], 3[11]\n",
+    "    # Change low gain to 2 for indexing purposes.\n",
+    "    g[g==3] = 2\n",
+    "\n",
+    "    if replace_wrong_gain_value > -1:\n",
+    "        g = replace_pixels_gain_values(\n",
+    "            g,\n",
+    "            replace_wrong_gain_value,\n",
+    "            wrong_gain_pixels,\n",
+    "            local_karabo_da,\n",
+    "        )\n",
+    "\n",
     "    # Select memory cells\n",
     "    if memory_cells > 1:\n",
     "        \"\"\"\n",
@@ -418,20 +430,6 @@
     "        offset_map_cell = offset_map\n",
     "        mask_cell = mask\n",
     "\n",
-    "    # Jungfrau gains 0[00], 1[01], 3[11]\n",
-    "    # Change low gain to 2 for indexing purposes.\n",
-    "    g[g==3] = 2\n",
-    "\n",
-    "    if replace_wrong_gain_value > -1:\n",
-    "        g = replace_pixels_gain_values(\n",
-    "            g,\n",
-    "            replace_wrong_gain_value,\n",
-    "            wrong_gain_pixels,\n",
-    "            local_karabo_da,\n",
-    "            mask_cell,\n",
-    "            memory_cells,\n",
-    "        )\n",
-    "\n",
     "    # Offset correction\n",
     "    offset = np.choose(g, np.moveaxis(offset_map_cell, -1, 0))\n",
     "\n",
-- 
GitLab