Remove spurious squeeze

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)