Skip to content

Commit 97f2811

Browse files
committed
COMP: UnboundLocalError in _GetImageFromArray
To address: UnboundLocalError: local variable 'ImageType' referenced before assignment with Python 3.7.
1 parent 1f5f9b8 commit 97f2811

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

Wrapping/Generators/Python/itkExtras.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -268,6 +268,8 @@ def _GetImageFromArray(arr, function, is_vector):
268268
raise ImportError('Numpy not available.')
269269
import itk
270270
PixelType = _get_itk_pixelid(arr)
271+
Dimension = arr.ndim
272+
ImageType = itk.Image[PixelType, Dimension]
271273
if is_vector:
272274
Dimension = arr.ndim - 1
273275
if arr.flags['C_CONTIGUOUS']:
@@ -281,9 +283,6 @@ def _GetImageFromArray(arr, function, is_vector):
281283
ImageType = itk.Image[ itk.RGBAPixel[itk.UC], Dimension ]
282284
else:
283285
ImageType = itk.Image[ itk.Vector[PixelType, VectorDimension] , Dimension]
284-
else:
285-
Dimension = arr.ndim
286-
ImageType = itk.Image[PixelType, Dimension]
287286
templatedFunction = getattr(itk.PyBuffer[ImageType], function)
288287
return templatedFunction(arr, is_vector)
289288

0 commit comments

Comments
 (0)