diff --git a/arraycontext/impl/pytato/utils.py b/arraycontext/impl/pytato/utils.py index 0af542041b1d4d022f1f5a311185461e1a0a6c71..a5582d189bcca017b9ddbb84424d43cd17d2ef55 100644 --- a/arraycontext/impl/pytato/utils.py +++ b/arraycontext/impl/pytato/utils.py @@ -23,7 +23,7 @@ THE SOFTWARE. """ -from typing import TYPE_CHECKING, Any, Dict, Mapping, Optional, Set, Tuple +from typing import TYPE_CHECKING, Any, Dict, Mapping, Set, Tuple from pytato.array import ( AbstractResultWithNamedArrays, @@ -118,7 +118,7 @@ class ArgSizeLimitingPytatoLoopyPyOpenCLTarget(LoopyPyOpenCLTarget): self.limit_arg_size_nbytes = limit_arg_size_nbytes @memoize_method - def get_loopy_target(self) -> Optional["lp.PyOpenCLTarget"]: + def get_loopy_target(self) -> "lp.PyOpenCLTarget": from loopy import PyOpenCLTarget return PyOpenCLTarget(limit_arg_size_nbytes=self.limit_arg_size_nbytes)