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

interpolate if necessary

Browse source
This commit is contained in:
Marcell Vazquez-Chanlatte 2016-10-10 22:10:56 -07:00
parent 0781605f78
commit 2ade1a678b
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
robustness.py
View file

@ -3,6 +3,7 @@
from functools import singledispatch from functools import singledispatch
from operator import sub, add from operator import sub, add
import numpy as np
from lenses import lens from lenses import lens
import stl.ast import stl.ast
@ -52,6 +53,7 @@ op_lookup = {
} }
@pointwise_robustness.register(stl.LinEq) @pointwise_robustness.register(stl.LinEq)
def _(stl): def _(stl):
op = op_lookup[stl.op] op = op_lookup[stl.op]
@ -64,4 +66,5 @@ def eval_terms(lineq, x, t):
def eval_term(x, t): def eval_term(x, t):
return lambda term: term.coeff*x[term.id.name][t] # TODO(lift interpolation much higher)
return lambda term: term.coeff*np.interp(t, x.index, x[term.id.name])