From aff9ab929bd156b9ba62cd2fdadb6c4d867db3c3 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 6 Aug 2024 12:31:42 -0500
Subject: [PATCH] Fix a return type in ArgSizeLimitingPytatoLoopyPyOpenCLTarget
---
arraycontext/impl/pytato/utils.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/arraycontext/impl/pytato/utils.py b/arraycontext/impl/pytato/utils.py
index 0af5420..a5582d1 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)
--
GitLab