Skip to content

Commit

Permalink
fix: bad assumption on callable control input function
Browse files Browse the repository at this point in the history
  • Loading branch information
briandesilva committed Sep 29, 2020
1 parent 0f7ca74 commit 9d58553
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pysindy/pysindy.py
Original file line number Diff line number Diff line change
@@ -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]

0 comments on commit 9d58553

Please sign in to comment.