Skip to content

Commit 94134d2

Browse files
Anna Störikoaseyboldt
authored andcommitted
Fix bug due to aesara moved import
1 parent c52f330 commit 94134d2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sunode/wrappers/as_theano.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,7 @@ def grad(self, inputs, g):
218218
return [
219219
aet.zeros_like(inputs[0]),
220220
aet.sum(g[:, None, :] * sens, (0, -1)),
221-
aet.grad_not_implemented(self, 2, inputs[-1])
221+
aesara.gradient.grad_not_implemented(self, 2, inputs[-1])
222222
]
223223

224224

@@ -257,7 +257,7 @@ def grad(self, inputs, g):
257257
y0, params, params_fixed = inputs
258258
backward = SolveODEAdjointBackward(self._solver, self._t0, self._tvals)
259259
lamda, gradient = backward(y0, params, params_fixed, g)
260-
return [-lamda, gradient, aet.grad_not_implemented(self, 2, params_fixed)]
260+
return [-lamda, gradient, aesara.gradient.grad_not_implemented(self, 2, params_fixed)]
261261

262262

263263
class SolveODEAdjointBackward(Op):

0 commit comments

Comments
 (0)