diff --git a/arraycontext/container/traversal.py b/arraycontext/container/traversal.py index c9ac2d545012fe5921ca9fd501a03212c34f955e..99b74a956b76ae6e6aba32a64fe75257e1b3d830 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): + def _tuple_if_not_tuple(x: Any) -> tuple: if not isinstance(x, tuple): assert isinstance(x, (str, int)) return x,