From 1bc50483354877ae1370a10296d37fdaa1970622 Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Fri, 11 Jun 2021 18:26:44 -0500
Subject: [PATCH] fixup! adds keyed counterparts to ((rec)?)map_array_container
- fixes function types, thereby passing mypy CI
---
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 f8dc5f2..f8a5411 100644
--- a/arraycontext/container/traversal.py
+++ b/arraycontext/container/traversal.py
@@ -255,7 +255,7 @@ def keyed_map_array_container(
def _keyed_map_array_container_impl(
- f: Callable[[Any], Any],
+ f: Callable[[Tuple[Union[str, int], ...], Any], Any],
ary: ArrayContainerT, *,
leaf_cls: Optional[type] = None,
recursive: bool = False) -> ArrayContainerT:
--
GitLab