BUG: Add missing xarray-related imports

Author: thewtex

itkwidgets/integrations/__init__.py

@@ -138,10 +138,12 @@ def _get_viewer_point_sets(point_sets):
         import torch
         if isinstance(point_sets, torch.Tensor):
             return point_sets.numpy()
-    if isinstance(point_sets, xr.DataArray):
-        return xarray_data_array_to_numpy(point_sets)
-    if isinstance(point_sets, xr.Dataset):
-        return xarray_data_set_to_numpy(point_sets)
+    if HAVE_XARRAY:
+        import xarray as xr
+        if isinstance(point_sets, xr.DataArray):
+            return xarray_data_array_to_numpy(point_sets)
+        if isinstance(point_sets, xr.Dataset):
+            return xarray_data_set_to_numpy(point_sets)
     return point_sets
 
 
@@ -151,6 +153,7 @@ def _detect_render_type(data, input_type) -> RenderType:
     elif isinstance(data, itkwasm.PointSet):
         return RenderType.POINT_SET
     elif HAVE_MULTISCALE_SPATIAL_IMAGE:
+        from multiscale_spatial_image import MultiscaleSpatialImage
         if isinstance(data, MultiscaleSpatialImage):
             return RenderType.IMAGE
     elif isinstance(data, (zarr.Array, zarr.Group)):
@@ -190,11 +193,13 @@ def _detect_render_type(data, input_type) -> RenderType:
                 return RenderType.POINT_SET
             else:
                 return RenderType.IMAGE
-    if isinstance(data, xr.DataArray):
-        if input_type == 'point_sets':
-            return RenderType.POINT_SET
-        else:
-            return RenderType.IMAGE
+    if HAVE_XARRAY:
+        import xarray as xr
+        if isinstance(data, xr.DataArray):
+            if input_type == 'point_sets':
+                return RenderType.POINT_SET
+            else:
+                return RenderType.IMAGE
     if isinstance(data, xr.Dataset):
         if input_type == 'point_sets':
             return RenderType.POINT_SET