From 8ca80dd04bacdc5a4869d6d52ade61ef50c1e2b8 Mon Sep 17 00:00:00 2001 From: Robert Haase Date: Sun, 29 Sep 2024 09:46:03 +0200 Subject: [PATCH] adding pyclesperanto Array to Image type --- pyclesperanto_prototype/_tier0/_types.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/pyclesperanto_prototype/_tier0/_types.py b/pyclesperanto_prototype/_tier0/_types.py index 7941d18c..474dd70a 100644 --- 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 \