Commit ac34312c authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Fix tests (broken from renaming autodiff submodule)

parent ea6e0238
Pipeline #17174 failed with stage
in 2 minutes and 52 seconds
......@@ -6,7 +6,7 @@ from pystencils_autodiff._field_to_tensors import ( # NOQA
tf_variable_from_field, torch_tensor_from_field)
from pystencils_autodiff._adjoint_field import AdjointField
from pystencils_autodiff._autodiff import (
AutoDiffAstPair, AutoDiffOp, create_backward_assignments, get_jacobian_of_assignments)
AutoDiffAstPair, AutoDiffOp, create_backward_assignments, get_jacobian_of_assignments, DiffModes)
__all__ = ['backends',
'AdjointField',
......@@ -16,7 +16,7 @@ __all__ = ['backends',
'AutoDiffAstPair',
"tf_constant_from_field", " tf_placeholder_from_field",
"tf_scalar_variable_from_field", " tf_variable_from_field",
"torch_tensor_from_field"]
"torch_tensor_from_field", "DiffModes"]
sys.modules['pystencils.autodiff'] = pystencils_autodiff
sys.modules['pystencils.autodiff.backends'] = pystencils_autodiff.backends
......@@ -2,7 +2,8 @@ import sympy as sp
import pystencils as ps
import pystencils_autodiff
from pystencils_autodiff.autodiff import DiffModes
from pystencils_autodiff import DiffModes
def test_simple_2d_check_assignment_collection():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment