Skip to content

Commit c311c36

Browse files
committed
Fix #241
1 parent 28332a3 commit c311c36

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

Python/tigre/utilities/geometry.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,10 @@ def __check_and_repmat__(self, attrib, angles):
123123

124124
elif type(old_attrib) == np.ndarray:
125125
if old_attrib.ndim == 1:
126-
if old_attrib.shape in [(3,), (2,), (1,)]:
126+
if old_attrib.shape in [(3,), (2,)] and not attrib in ['DSD', 'DSO', 'COR']:
127+
new_attrib = matlib.repmat(old_attrib, angles.shape[0], 1)
128+
setattr(self, attrib, new_attrib)
129+
elif old_attrib.shape in [(1,)]:
127130
new_attrib = matlib.repmat(old_attrib, angles.shape[0], 1)
128131
setattr(self, attrib, new_attrib)
129132
elif old_attrib.shape == (angles.shape[0],):

0 commit comments

Comments
 (0)