fix: bad assumption on callable control input function

dynamicslab · Sep 29, 2020 · 9d58553 · 9d58553
1 parent 0f7ca74
commit 9d58553
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]