From 1b460588e915a23f34e80d3363dd8a4ef7cf1dcc Mon Sep 17 00:00:00 2001
From: David Hammer <dhammer@mailbox.org>
Date: Thu, 10 Feb 2022 15:29:54 +0100
Subject: [PATCH] Remove spurious squeeze
---
src/calng/base_gpu.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/calng/base_gpu.py b/src/calng/base_gpu.py
index c0619d0a..333eb053 100644
--- a/src/calng/base_gpu.py
+++ b/src/calng/base_gpu.py
@@ -126,7 +126,7 @@ class BaseGpuRunner:
return self.reshaped_data_gpu.get(out=out)
def load_data(self, raw_data):
- self.input_data_gpu.set(np.squeeze(raw_data))
+ self.input_data_gpu.set(raw_data)
def load_cell_table(self, cell_table):
self.cell_table_gpu.set(cell_table)
--
GitLab