From e340a2906cbd7e1eb57100593e1f6a082790667b Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Sat, 12 Jun 2021 22:38:01 -0500
Subject: [PATCH] makes type of _tuple_if_not_tuple more precise
---
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 99b74a9..c0fede9 100644
--- a/arraycontext/container/traversal.py
+++ b/arraycontext/container/traversal.py
@@ -271,7 +271,7 @@ def _keyed_map_array_container_impl(
if type(_ary) is leaf_cls: # type(ary) is never None
return f(keys, _ary)
elif is_array_container(_ary):
- def _tuple_if_not_tuple(x: Any) -> tuple:
+ def _tuple_if_not_tuple(x: Any) -> Tuple[Union[str, int], ...]:
if not isinstance(x, tuple):
assert isinstance(x, (str, int))
return x,
--
GitLab