teaching/mtl-aas
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

is check for TOP and BOT is flatten_binary op

Browse source
This commit is contained in:
Marcell Vazquez-Chanlatte 2017-02-27 22:16:01 -08:00
parent ff0253438d
commit 58c33f530c
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
stl/ast.py
View file

@ -14,9 +14,8 @@ t_sym = Symbol('t', positive=True)
def flatten_binary(phi, op, dropT, shortT): def flatten_binary(phi, op, dropT, shortT):
f = lambda x: x.args if isinstance(x, op) else [x] f = lambda x: x.args if isinstance(x, op) else [x]
args = [arg for arg in phi.args if not isinstance(arg, type(dropT))] args = [arg for arg in phi.args if arg is not dropT]
if any(arg is shortT for arg in args):
if any(isinstance(arg, type(shortT)) for arg in args):
return shortT return shortT
elif not args: elif not args:
return dropT return dropT