From 58c33f530c84b39a3dac809eaff4e7c6d40f1865 Mon Sep 17 00:00:00 2001 From: Marcell Vazquez-Chanlatte Date: Mon, 27 Feb 2017 22:16:01 -0800 Subject: [PATCH] is check for TOP and BOT is flatten_binary op --- stl/ast.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/stl/ast.py b/stl/ast.py index 2a521c4..1721859 100644 --- a/stl/ast.py +++ b/stl/ast.py @@ -14,9 +14,8 @@ t_sym = Symbol('t', positive=True) def flatten_binary(phi, op, dropT, shortT): f = lambda x: x.args if isinstance(x, op) else [x] - args = [arg for arg in phi.args if not isinstance(arg, type(dropT))] - - if any(isinstance(arg, type(shortT)) for arg in args): + args = [arg for arg in phi.args if arg is not dropT] + if any(arg is shortT for arg in args): return shortT elif not args: return dropT