diff --git a/arraycontext/__init__.py b/arraycontext/__init__.py index bfbc14f720e8953bb4021317c28567c76b7df964..b8287cf1e54ef1071151499818183f9d0c2d4c78 100644 --- a/arraycontext/__init__.py +++ b/arraycontext/__init__.py @@ -29,7 +29,9 @@ THE SOFTWARE. """ import sys -from .context import ArrayContext, Array, Scalar, tag_axes +from .context import ( + Array, ArrayT, + ArrayContext, Scalar, tag_axes) from .transform_metadata import (CommonSubexpressionTag, ElementwiseMapKernelTag) @@ -38,6 +40,7 @@ from .transform_metadata import (CommonSubexpressionTag, from .metadata import _FirstAxisIsElementsTag from .container import ( + ArrayOrContainerT as ArrayOrContainer, ArrayOrContainerT, ArrayContainer, NotAnArrayContainerError, is_array_container, is_array_container_type, get_container_context_opt, @@ -78,11 +81,13 @@ from .loopy import make_loopy_program __all__ = ( - "ArrayContext", "Scalar", "Array", "tag_axes", + "Array", "ArrayT", + "ArrayContext", "Scalar", "tag_axes", "CommonSubexpressionTag", "ElementwiseMapKernelTag", + "ArrayOrContainer", "ArrayOrContainerT", "ArrayContainer", "NotAnArrayContainerError", "is_array_container", "is_array_container_type", "get_container_context_opt",