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

implement paramater setting

Browse source
This commit is contained in:
Marcell Vazquez-Chanlatte 2016-10-08 20:42:32 -07:00
parent e3ff61e971
commit 0b69607924
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
utils.py
View file

@ -1,6 +1,6 @@
from collections import deque from collections import deque
from lenses import lens from lenses import lens, Lens
import funcy as fn import funcy as fn
import sympy import sympy
@ -69,3 +69,8 @@ def param_lens(phi):
return [lens().const] if isinstance(leaf, LinEq) else [lens().interval[0], lens().interval[1]] return [lens().const] if isinstance(leaf, LinEq) else [lens().interval[0], lens().interval[1]]
return ast_lens(phi, pred=type_pred(LinEq, F, G), focus_lens=focus_lens).filter_(is_sym) return ast_lens(phi, pred=type_pred(LinEq, F, G), focus_lens=focus_lens).filter_(is_sym)
def set_params(stl_or_lens, val):
l = stl_or_lens if isinstance(stl_or_lens, Lens) else param_lens(stl_or_lens)
return l.modify(lambda x: val[str(x)] if str(x) in val else x)