diff --git a/pymbolic/interop/ast.py b/pymbolic/interop/ast.py index 3b6d4fdbf4e7029d01be8d3813f01a1f061fbebc..d7787b83c0a977ba1ed8c7caf50d3856f4a0ef95 100644 --- a/pymbolic/interop/ast.py +++ b/pymbolic/interop/ast.py @@ -104,9 +104,9 @@ class ASTToPymbolic(ASTMapper): return p.Product((x, y)) bin_op_map = { - ast.Add: ASTToPymbolic._add, - ast.Sub: ASTToPymbolic._sub, - ast.Mult: ASTToPymbolic._mult, + ast.Add: _add, + ast.Sub: _sub, + ast.Mult: _mult, # MatMult ast.Div: p.Quotient, ast.FloorDiv: p.FloorDiv, @@ -135,10 +135,10 @@ class ASTToPymbolic(ASTMapper): return p.Product((-1, x),) unary_op_map = { - ast.Invert: ASTToPymbolic._neg, + ast.Invert: _neg, ast.Not: p.LogicalNot, # ast.UAdd: - ast.USub: ASTToPymbolic._neg, + ast.USub: _neg, } def map_UnaryOp(self, expr): # noqa