Skip to content

Commit a49ade3

Browse files
committed
add TODO refactor newton solver to use fwd_solver
1 parent 0121ec3 commit a49ade3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pytensor/optimise/fixed_point.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ def newton_solver(x_prev, *args, func, tol):
3030
grad = pt.linalg.solve(jac, f_root)
3131
x = x_prev - grad
3232

33+
# TODO: consider if this can all be done as a single call to `fwd_solver` as in the jax test case
3334
is_converged = _check_convergence(x - x_prev, tol)
3435

3536
return x, is_converged

0 commit comments

Comments
 (0)