From 2ade1a678b119201e573fea911d3e290d79c40c7 Mon Sep 17 00:00:00 2001 From: Marcell Vazquez-Chanlatte Date: Mon, 10 Oct 2016 22:10:56 -0700 Subject: [PATCH] interpolate if necessary --- robustness.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/robustness.py b/robustness.py index 9b9e280..058dd0a 100644 --- a/robustness.py +++ b/robustness.py @@ -3,6 +3,7 @@ from functools import singledispatch from operator import sub, add +import numpy as np from lenses import lens import stl.ast @@ -52,6 +53,7 @@ op_lookup = { } + @pointwise_robustness.register(stl.LinEq) def _(stl): op = op_lookup[stl.op] @@ -64,4 +66,5 @@ def eval_terms(lineq, 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])