diff --git a/pymbolic/interop/sympy.py b/pymbolic/interop/sympy.py index a9626cf3bed434fce1faf74a3c715384dc4cf8fe..b4807712591e695a434662d4fc93b88f3322575d 100644 --- a/pymbolic/interop/sympy.py +++ b/pymbolic/interop/sympy.py @@ -67,7 +67,7 @@ class SympyToPymbolicMapper(SympyLikeToPymbolicMapper): # We only handle piecewises with 2 arguments! assert len(expr.args) == 2 # We only handle if/else cases - assert expr.args[0][1] is True + assert expr.args[1][1].is_Boolean and bool(expr.args[1][1]) is True then = self.rec(expr.args[0][0]) else_ = self.rec(expr.args[1][0]) cond = self.rec(expr.args[0][1])