From e044a9cf76bdcaaf76d74cb748f2da7fa893b558 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 28 Feb 2020 13:36:41 +0100
Subject: [PATCH] Fix test custom printer
---
tests/test_module_printing.py | 6 +-----
1 file changed, 1 insertion(+), 5 deletions(-)
diff --git a/tests/test_module_printing.py b/tests/test_module_printing.py
index ad3c20a..127ec95 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))
--
GitLab