Skip to content
Snippets Groups Projects
Commit d700df44 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Re-export ArrayWrapper, coerce_to_field to top-level

parent 8abd3aab
Branches
Tags
No related merge requests found
......@@ -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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment