From ac34312c1a924f3260ef57f142bfa8a87a1a839a Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 13 Aug 2019 12:25:45 +0200
Subject: [PATCH] Fix tests (broken from renaming autodiff submodule)
---
src/pystencils_autodiff/__init__.py | 4 ++--
tests/test_autodiff.py | 3 ++-
2 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/src/pystencils_autodiff/__init__.py b/src/pystencils_autodiff/__init__.py
index fafab5a..4a70889 100644
--- a/src/pystencils_autodiff/__init__.py
+++ b/src/pystencils_autodiff/__init__.py
@@ -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
diff --git a/tests/test_autodiff.py b/tests/test_autodiff.py
index 3ddc3a7..8d3f535 100644
--- a/tests/test_autodiff.py
+++ b/tests/test_autodiff.py
@@ -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():
--
GitLab