From 2f88897c00e156cbdd0ddf09e1989255a9905f2f Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Mon, 16 Aug 2021 13:57:25 -0500
Subject: [PATCH] expose DataWrapper/InputArgumentBase under pytato namespace

---
 pytato/__init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pytato/__init__.py b/pytato/__init__.py
index 4a89b01..6f4d18a 100644
--- a/pytato/__init__.py
+++ b/pytato/__init__.py
@@ -26,7 +26,7 @@ THE SOFTWARE.
 
 from pytato.array import (
         Array, AbstractResultWithNamedArrays, DictOfNamedArrays, Placeholder,
-        IndexLambda, NamedArray,
+        IndexLambda, NamedArray, DataWrapper, InputArgumentBase,
 
         make_dict_of_named_arrays,
         make_placeholder, make_size_param, make_data_wrapper,
@@ -62,6 +62,7 @@ from pytato.visualization import get_dot_graph, show_dot_graph
 __all__ = (
         "Array", "AbstractResultWithNamedArrays", "DictOfNamedArrays",
         "Placeholder", "IndexLambda", "NamedArray", "LoopyCall",
+        "DataWrapper", "InputArgumentBase",
 
         "make_dict_of_named_arrays", "make_placeholder", "make_size_param",
         "make_data_wrapper", "einsum",
-- 
GitLab