diff --git a/arraycontext/__init__.py b/arraycontext/__init__.py index 12475db53b048b3d1dfa93faa98fb7f42eb446ea..b367382c834aa3c1a784cef8e4467475bd35e592 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 7fd739b1b3370289666e6537a8f4f14fec59b420..b5d91a9ba69b33de5764159ddfd3b4fd51ea69c8 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 = {}