diff --git a/pystencils/astnodes.py b/pystencils/astnodes.py index 6976b857c1757734472ff63d344a2bf63220b0f1..ef0bcc6d758fdb67fc42b708038882360a9eee65 100644 --- a/pystencils/astnodes.py +++ b/pystencils/astnodes.py @@ -377,8 +377,6 @@ class Block(Node): return tmp def replace(self, child, replacements): - if self._nodes.count(child) != 1: - print('here') assert self._nodes.count(child) == 1 idx = self._nodes.index(child) del self._nodes[idx] diff --git a/pystencils/transformations.py b/pystencils/transformations.py index c022e728db0c7df47368be26941842e9664b2c76..2113c84483a4c15e67e971860b285a2cc156f533 100644 --- a/pystencils/transformations.py +++ b/pystencils/transformations.py @@ -773,7 +773,7 @@ def simplify_conditionals(node: ast.Node, loop_counter_simplification: bool = Fa """ for conditional in node.atoms(ast.Conditional): # TODO simplify conditional before the type system! Casts make it very hard here - # conditional.condition_expr = sp.simplify(conditional.condition_expr) + conditional.condition_expr = sp.simplify(conditional.condition_expr) if conditional.condition_expr == sp.true: conditional.parent.replace(conditional, [conditional.true_block]) elif conditional.condition_expr == sp.false: