From d700df44d34aebe6111623502f43045c9f4c64f5 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 19 Dec 2019 10:54:57 +0100
Subject: [PATCH] Re-export ArrayWrapper, coerce_to_field to top-level
---
src/pystencils_autodiff/__init__.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/src/pystencils_autodiff/__init__.py b/src/pystencils_autodiff/__init__.py
index e15b7d2..ea06911 100644
--- a/src/pystencils_autodiff/__init__.py
+++ b/src/pystencils_autodiff/__init__.py
@@ -7,6 +7,7 @@ from pystencils_autodiff._adjoint_field import AdjointField
from pystencils_autodiff._autodiff import (
AutoDiffAstPair, AutoDiffBoundaryHandling, AutoDiffOp, DiffModes, create_backward_assignments,
get_jacobian_of_assignments)
+from pystencils_autodiff.field_tensor_conversion import ArrayWrapper, coerce_to_field
__all__ = ['backends',
'AdjointField',
@@ -16,7 +17,9 @@ __all__ = ['backends',
'AutoDiffAstPair',
'tensorflow_jit',
'DiffModes',
- 'AutoDiffBoundaryHandling']
+ 'AutoDiffBoundaryHandling',
+ 'ArrayWrapper',
+ 'coerce_to_field']
sys.modules['pystencils.autodiff'] = pystencils_autodiff
sys.modules['pystencils.autodiff.backends'] = pystencils_autodiff.backends
--
GitLab