adding pyclesperanto Array to Image type

clEsperanto · Sep 29, 2024 · 8ca80dd · 8ca80dd
1 parent 620ae07
commit 8ca80dd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pyclesperanto_prototype/_tier0/_types.py b/pyclesperanto_prototype/_tier0/_types.py
@@ -3,7 +3,12 @@
 import pyopencl as cl
 from typing import Union
 
-Image = Union[np.ndarray, OCLArray, cl.Image, _OCLImage]
+try:
+    import pyclesperanto
+    Image = Union[np.ndarray, OCLArray, cl.Image, _OCLImage, pyclesperanto._pyclesperanto._Array]
+except ImportError:
+    Image = Union[np.ndarray, OCLArray, cl.Image, _OCLImage]
+
 
 def is_image(object):
     return isinstance(object, np.ndarray) or \