From 329678940d0670a9a99b93658951703a4d74cd69 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Mon, 28 Jun 2021 11:58:21 -0500
Subject: [PATCH] add keyed* to doc

---
 arraycontext/__init__.py            | 2 ++
 arraycontext/container/traversal.py | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/arraycontext/__init__.py b/arraycontext/__init__.py
index a338059..8562f7b 100644
--- a/arraycontext/__init__.py
+++ b/arraycontext/__init__.py
@@ -52,6 +52,7 @@ from .container.traversal import (
         rec_multimap_array_container,
         mapped_over_array_containers,
         multimapped_over_array_containers,
+        keyed_map_array_container, rec_keyed_map_array_container,
         thaw, freeze,
         from_numpy, to_numpy)
 
@@ -83,6 +84,7 @@ __all__ = (
         "rec_map_array_container", "rec_multimap_array_container",
         "mapped_over_array_containers",
         "multimapped_over_array_containers",
+        "keyed_map_array_container", "rec_keyed_map_array_container",
         "thaw", "freeze",
         "from_numpy", "to_numpy",
 
diff --git a/arraycontext/container/traversal.py b/arraycontext/container/traversal.py
index 1dc0d3b..0bd0a1d 100644
--- a/arraycontext/container/traversal.py
+++ b/arraycontext/container/traversal.py
@@ -7,6 +7,8 @@
 .. autofunction:: multimap_array_container
 .. autofunction:: rec_map_array_container
 .. autofunction:: rec_multimap_array_container
+.. autofunction:: keyed_map_array_container
+.. autofunction:: rec_keyed_map_array_container
 
 Traversing decorators
 ~~~~~~~~~~~~~~~~~~~~~
-- 
GitLab