From 7199da2b840a46e73f5beb582a310ba115b04547 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Sun, 23 Apr 2023 00:29:00 -0500 Subject: [PATCH] Fix signature of _normalize_pt_expr after pytato typing made precise --- arraycontext/impl/pytato/utils.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/arraycontext/impl/pytato/utils.py b/arraycontext/impl/pytato/utils.py index 2a3be3f..c014a93 100644 --- a/arraycontext/impl/pytato/utils.py +++ b/arraycontext/impl/pytato/utils.py @@ -26,8 +26,8 @@ THE SOFTWARE. from typing import TYPE_CHECKING, Any, Dict, Mapping, Optional, Set, Tuple from pytato.array import ( - Array, Axis as PtAxis, DataWrapper, DictOfNamedArrays, Placeholder, SizeParam, - make_placeholder) + AbstractResultWithNamedArrays, Array, Axis as PtAxis, DataWrapper, + DictOfNamedArrays, Placeholder, SizeParam, make_placeholder) from pytato.target.loopy import LoopyPyOpenCLTarget from pytato.transform import CopyMapper from pytools import UniqueNameGenerator, memoize_method @@ -78,8 +78,9 @@ class _DatawrapperToBoundPlaceholderMapper(CopyMapper): " DatawrapperToBoundPlaceholderMapper.") -def _normalize_pt_expr(expr: DictOfNamedArrays) -> Tuple[DictOfNamedArrays, - Mapping[str, Any]]: +def _normalize_pt_expr( + expr: DictOfNamedArrays + ) -> Tuple[AbstractResultWithNamedArrays, Mapping[str, Any]]: """ Returns ``(normalized_expr, bound_arguments)``. *normalized_expr* is a normalized form of *expr*, with all instances of -- GitLab