Fix #241

CERN · Jan 28, 2021 · c311c36 · c311c36
1 parent 28332a3
commit c311c36
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Python/tigre/utilities/geometry.py b/Python/tigre/utilities/geometry.py
@@ -123,7 +123,10 @@ def __check_and_repmat__(self, attrib, angles):
 
         elif type(old_attrib) == np.ndarray:
             if old_attrib.ndim == 1:
-                if old_attrib.shape in [(3,), (2,), (1,)]:
+                if old_attrib.shape in [(3,), (2,)] and not attrib in ['DSD', 'DSO', 'COR']:
+                    new_attrib = matlib.repmat(old_attrib, angles.shape[0], 1)
+                    setattr(self, attrib, new_attrib)
+                elif old_attrib.shape in [(1,)]:
                     new_attrib = matlib.repmat(old_attrib, angles.shape[0], 1)
                     setattr(self, attrib, new_attrib)
                 elif old_attrib.shape == (angles.shape[0],):