From e08969f6264b2a6c504596ec4898d226d32b4b61 Mon Sep 17 00:00:00 2001 From: Matt Smith Date: Thu, 26 Aug 2021 12:10:30 -0500 Subject: [PATCH] fix type signature of wrapper func Co-authored-by: Alex Fikl --- arraycontext/container/traversal.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/arraycontext/container/traversal.py b/arraycontext/container/traversal.py index ed6dc0c..b2abc74 100644 --- a/arraycontext/container/traversal.py +++ b/arraycontext/container/traversal.py @@ -142,7 +142,7 @@ def _multimap_array_container_impl( if len(container_indices) == 1 and reduce_func is None: # NOTE: if we just have one ArrayContainer in args, passing it through # _map_array_container_impl should be faster - def wrapper(ary: ContainerT) -> ContainerT: + def wrapper(ary: ArrayOrContainerT) -> ArrayOrContainerT: new_args = list(args) new_args[container_indices[0]] = ary return f(*new_args) -- GitLab