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

Start using lens.Recur

Browse source
This commit is contained in:
Marcell Vazquez-Chanlatte 2017-11-30 17:47:52 -08:00
parent 38b8c1838f
commit d9f14b0be4
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
stl/ast.py
View file

@ -71,7 +71,7 @@ class AST(object):
@property @property
def lineqs(self): def lineqs(self):
return set(lineq_lens(self).Each().collect()) return set(lineq_lens.collect()(self))
@property @property
def atomic_predicates(self): def atomic_predicates(self):
@ -340,6 +340,7 @@ def type_pred(*args):
return lambda x: type(x) in ast_types return lambda x: type(x) in ast_types
lineq_lens = fn.partial(ast_lens, pred=type_pred(LinEq), getter=True) # lineq_lens = fn.partial(ast_lens, pred=type_pred(LinEq), getter=True)
lineq_lens = lens.Recur(LinEq)
AP_lens = fn.partial(ast_lens, pred=type_pred(AtomicPred), getter=True) AP_lens = fn.partial(ast_lens, pred=type_pred(AtomicPred), getter=True)
and_or_lens = fn.partial(ast_lens, pred=type_pred(And, Or), getter=True) and_or_lens = fn.partial(ast_lens, pred=type_pred(And, Or), getter=True)