diff --git a/tests/test_module_printing.py b/tests/test_module_printing.py index d94bbef11ea44ad08110ad82d4a46010c3b9ad63..ad3c20a16e04b1376b342e8d70efe273b9ddd35b 100644 --- a/tests/test_module_printing.py +++ b/tests/test_module_printing.py @@ -92,13 +92,9 @@ def test_module_printing_globals(): 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, target) forward_ast.function_name = 'forward' - backward_ast = pystencils.create_kernel(backward_assignments, target) - backward_ast.function_name = 'backward' - module = TorchModule("hallo", [forward_ast, backward_ast]) + module = TorchModule("hallo", [forward_ast]) print(module) @@ -122,4 +118,3 @@ def test_custom_printer(): backward_ast.function_name = 'backward' module = TorchModule("hallo", [forward_ast, backward_ast]) print(DoesNotLikeTorchPrinter()(module)) -