diff --git a/tests/test_module_printing.py b/tests/test_module_printing.py index ad3c20a16e04b1376b342e8d70efe273b9ddd35b..127ec9563b4fbf336855723164ad849bf8fcdd09 100644 --- a/tests/test_module_printing.py +++ b/tests/test_module_printing.py @@ -110,11 +110,7 @@ def test_custom_printer(): z[0, 0]: x[0, 0] * sympy.log(TextureCachedField(x).at(sympy.Matrix((0.43, 3))) * y[0, 0]) }) - backward_assignments = create_backward_assignments(forward_assignments) - forward_ast = pystencils.create_kernel(forward_assignments) forward_ast.function_name = 'forward' - backward_ast = pystencils.create_kernel(backward_assignments) - backward_ast.function_name = 'backward' - module = TorchModule("hallo", [forward_ast, backward_ast]) + module = TorchModule("hallo", [forward_ast]) print(DoesNotLikeTorchPrinter()(module))