From 9d5855342e58651459aa884fde5f534d8b28661f Mon Sep 17 00:00:00 2001 From: briandesilva Date: Tue, 29 Sep 2020 14:44:27 -0700 Subject: [PATCH] fix: bad assumption on callable control input function --- pysindy/pysindy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pysindy/pysindy.py b/pysindy/pysindy.py index 752acedbe..a986003fd 100644 --- a/pysindy/pysindy.py +++ b/pysindy/pysindy.py @@ -735,7 +735,7 @@ def rhs(x, t): else: u_fun = u - if ndim(u_fun(1)) == 1: + if ndim(u_fun(t[0])) == 1: def rhs(x, t): return self.predict(x[newaxis, :], u_fun(t).reshape(1, -1))[0]