Fix enqueue_fill signature

inducer · Aug 5, 2022 · 7ef5ce5 · 7ef5ce5
1 parent d642040
commit 7ef5ce5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pyopencl/__init__.py b/pyopencl/__init__.py
@@ -22,7 +22,7 @@
 
 from sys import intern
 from warnings import warn
-from typing import Union, Any
+from typing import Union, Any, Optional, Sequence
 
 from pyopencl.version import VERSION, VERSION_STATUS, VERSION_TEXT  # noqa
 
@@ -1821,8 +1821,10 @@ def enqueue_copy(queue, dest, src, **kwargs):
 
 # {{{ enqueue_fill
 
-def enqueue_fill(queue: CommandQueue, dest: Union[MemoryObjectHolder, SVMPointer],
-        pattern: Any, size: int, *, offset: int = 0, wait_for=None) -> Event:
+def enqueue_fill(queue: CommandQueue,
+        dest: "Union[MemoryObjectHolder, SVMPointer]",
+        pattern: Any, size: int, *, offset: int = 0,
+        wait_for: Optional[Sequence[Event]] = None) -> Event:
     """
     .. versionadded:: 2022.2
     """