Skip to content

Commit f6e0948

Browse files
PERF: Fix regression
1 parent 0d9700f commit f6e0948

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

pysindy/differentiation/finite_difference.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,11 @@ def _differentiate(
286286
s[self.axis] = slice(start, stop)
287287
interior = interior + x[tuple(s)] * coeffs[i]
288288
else:
289-
t = AxesArray(np.array(t), axes={"ax_time": 0, "ax_coord": 1})
289+
if not isinstance(t, AxesArray):
290+
t = np.asarray(t)
291+
if t.ndim == 1:
292+
t = t[:, np.newaxis]
293+
t = AxesArray(t, axes={"ax_time": 0, "ax_coord": 1})
290294
coeffs = self._coefficients(t)
291295
interior = self._accumulate(coeffs, x)
292296
s[self.axis] = slice((self.n_stencil - 1) // 2, -(self.n_stencil - 1) // 2)

0 commit comments

Comments
 (0)