diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py index 70b3742616da4c6b1da0f6efa217c5cfef46e7ff..8c09226fde8df846c6cfefb1af5ef8a47971e772 100644 --- a/arraycontext/impl/pytato.py +++ b/arraycontext/impl/pytato.py @@ -305,7 +305,7 @@ class PytatoArrayContext(ArrayContext): # }}} def compile(self, f: Callable[[Any], Any], - inputs_like: Tuple[Union[Number, np.array], ...]) -> Callable[..., Any]: + inputs_like: Tuple[Union[Number, np.ndarray], ...]) -> Callable[..., Any]: from pytools.obj_array import flat_obj_array from arraycontext.impl import _is_meshmode_dofarray from meshmode.dof_array import DOFArray