fix preload bug

3dem · Oct 18, 2023 · 9b4ad60 · 9b4ad60
1 parent 4cb7696
commit 9b4ad60
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 1 deletion.
diff --git a/dynamight/data/handlers/__pycache__/particle_dataset.cpython-310.pyc b/dynamight/data/handlers/__pycache__/particle_dataset.cpython-310.pyc
diff --git a/dynamight/data/handlers/particle_dataset.py b/dynamight/data/handlers/particle_dataset.py
@@ -112,7 +112,10 @@ def preload_images(self):
             this_file_mask = unique_reverse == file_idx
             this_file_stack_indices = self.part_stack_idx[this_file_mask]
             # Take slices of images for this data set
-            this_file_images = mrc.data[this_file_stack_indices]
+            if len(mrc.data.shape) > 2:
+                this_file_images = mrc.data[this_file_stack_indices]
+            else:
+                this_file_images = np.expand_dims(mrc.data, 0)
             # Particles indices with images in this file
             this_file_index_list = part_index_list[this_file_mask]
 

diff --git a/dynamight/deformations/__pycache__/optimize_deformations.cpython-310.pyc b/dynamight/deformations/__pycache__/optimize_deformations.cpython-310.pyc
diff --git a/dynamight/inverse_deformations/__pycache__/optimize_inverse_deformations.cpython-310.pyc b/dynamight/inverse_deformations/__pycache__/optimize_inverse_deformations.cpython-310.pyc