From 0980f4b2f98d9f7ce40acf001d67eb2024c2293d Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Fri, 18 Jun 2021 09:18:17 -0500
Subject: [PATCH] do not expose (rec_?)keyed_map_array_container under :mod:
---
arraycontext/__init__.py | 4 ----
arraycontext/impl/pytato.py | 8 ++++----
2 files changed, 4 insertions(+), 8 deletions(-)
diff --git a/arraycontext/__init__.py b/arraycontext/__init__.py
index 12475db..b367382 100644
--- a/arraycontext/__init__.py
+++ b/arraycontext/__init__.py
@@ -47,8 +47,6 @@ 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)
@@ -78,8 +76,6 @@ __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/impl/pytato.py b/arraycontext/impl/pytato.py
index 7fd739b..b5d91a9 100644
--- a/arraycontext/impl/pytato.py
+++ b/arraycontext/impl/pytato.py
@@ -195,8 +195,8 @@ class PytatoCompiledOperator:
def __call__(self, *args):
- from arraycontext import (rec_keyed_map_array_container,
- is_array_container)
+ from arraycontext.container.traversal import (rec_keyed_map_array_container,
+ is_array_container)
import pytato as pt
def to_arg_descr(arg):
@@ -295,8 +295,8 @@ class PytatoExecutable:
def __call__(self, *args):
import pytato as pt
import pyopencl.array as cla
- from arraycontext import (is_array_container,
- rec_keyed_map_array_container)
+ from arraycontext.container import is_array_container
+ from arraycontext.container.traversal import rec_keyed_map_array_container
input_kwargs_to_loopy = {}
--
GitLab