Merge pull request #1080 from thewtex/cupy-check

fix(itkwasm): fix check for cupy

Author: thewtex

packages/core/python/itkwasm/itkwasm/__init__.py

@@ -1,6 +1,6 @@
 """itkwasm: Python interface to itk-wasm WebAssembly modules."""
 
-__version__ = "1.0b168"
+__version__ = "1.0b169"
 
 from .interface_types import InterfaceTypes
 from .image import Image, ImageType

packages/core/python/itkwasm/itkwasm/to_cupy_array.py

@@ -7,7 +7,7 @@
 from numpy.typing import ArrayLike
 
 try:
-    distribution('cupy')
+    import cupy
     _CUPY_AVAILABLE = True
 except:
     _CUPY_AVAILABLE = False
@@ -50,4 +50,4 @@ def array_like_to_cupy_array(arr: ArrayLike) -> "cp.ndarray":
             arr = arr.compute()
         if isinstance(arr, cp.ndarray):
             return arr
-    return cp.array(arr)
+    return cp.array(arr)