From 52070129ecd1ec2efec1701b222eba3ecc11b36d Mon Sep 17 00:00:00 2001
From: David Hammer <dhammer@mailbox.org>
Date: Tue, 15 Feb 2022 13:01:59 +0100
Subject: [PATCH] Fix: drop singleton dimension for DSSC GPU kernel runner
 image shape

---
 src/calng/DsscCorrection.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/calng/DsscCorrection.py b/src/calng/DsscCorrection.py
index 4df05a25..86566075 100644
--- a/src/calng/DsscCorrection.py
+++ b/src/calng/DsscCorrection.py
@@ -232,7 +232,8 @@ class DsscCorrection(BaseCorrection):
                 return
 
         try:
-            self.kernel_runner.load_data(image_data)
+            # drop the singleton dimension for kernel runner
+            self.kernel_runner.load_data(image_data[:, 0])
         except ValueError as e:
             self.log_status_warn(f"Failed to load data: {e}")
             return
-- 
GitLab