fix type signature of wrapper func

Co-authored-by: Alex Fikl <alexfikl@gmail.com>

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)