ArrayContext.zeros: accept DTypeLike

Author: inducer

arraycontext/context.py

@@ -135,6 +135,7 @@
 
 if TYPE_CHECKING:
     import numpy as np
+    from numpy.typing import DTypeLike
 
     import loopy
     from pytools.tag import ToTagSetConvertible
@@ -223,7 +224,7 @@ def __hash__(self) -> int:
 
     def zeros(self,
               shape: int | tuple[int, ...],
-              dtype: np.dtype[Any]) -> Array:
+              dtype: DTypeLike) -> Array:
         warn(f"{type(self).__name__}.zeros is deprecated and will stop "
             "working in 2025. Use actx.np.zeros instead.",
             DeprecationWarning, stacklevel=2)