From a3e68ac9b9620c85a23227fa34febb73d7081de4 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 29 Jun 2022 14:25:40 -0500
Subject: [PATCH] Expose types for subpackages in package root
---
arraycontext/__init__.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/arraycontext/__init__.py b/arraycontext/__init__.py
index bfbc14f..b8287cf 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",
--
GitLab