BUG: to_itk_image understands itk.VectorImage

thewtex · thewtex · commit c7bd4ae7f36c · 2020-07-23T16:56:56.000-04:00
diff --git a/itkwidgets/_transform_types.py b/itkwidgets/_transform_types.py
@@ -232,7 +232,7 @@ def _numpy_array_to_point_set(point_set_like):
 
 def to_itk_image(image_like):
 
-    if isinstance(image_like, itk.Image):
+    if isinstance(image_like, (itk.Image, itk.VectorImage)):
         return image_like
 
     if is_arraylike(image_like):
