Skip to content

Commit

Permalink
fix: dependencies errors with scipy
Browse files Browse the repository at this point in the history
  • Loading branch information
Kleyt0n committed Jun 18, 2024
1 parent 779f146 commit 737aaca
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/optymus/methods/_second_order.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,7 @@ def penalized_obj(x):
if jnp.linalg.norm(g) < tol:
break

d = jax.scipy.linalg.solve(M, -g)
d = jnp.linalg.solve(M, -g)

r = line_search(f=penalized_obj, x=x, d=d, learning_rate=learning_rate)
x = r['xopt']
Expand Down

0 comments on commit 737aaca

Please sign in to comment.