Skip to content

Commit

Permalink
Merge branch 'main' into add-explicit-interface-for-solver-results-to…
Browse files Browse the repository at this point in the history
…-dynamics-backend
  • Loading branch information
DanPuzzuoli authored Jan 11, 2024
2 parents 09e8978 + e8c64fd commit 25ee09b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions qiskit_dynamics/solvers/solver_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ def _is_jax_method(method: any) -> bool:
def _is_diffrax_method(method: any) -> bool:
"""Check if method is a diffrax method."""
try:
from diffrax.solver import AbstractSolver
from diffrax import AbstractSolver

return isinstance(method, AbstractSolver)
except ImportError:
Expand Down Expand Up @@ -156,7 +156,7 @@ def solve_ode(
- ``'jax_RK4'``: JAX backend implementation of ``'RK4'`` method.
- ``'jax_odeint'``: Calls ``jax.experimental.ode.odeint`` variable step solver.
- ``diffrax.diffeqsolve`` - a JAX solver function, called by passing ``method``
as a valid ``diffrax.solver.AbstractSolver`` instance. Requires the ``diffrax`` library.
as a valid ``diffrax.AbstractSolver`` instance. Requires the ``diffrax`` library.
Results are returned as a :class:`OdeResult` object.
Expand Down

0 comments on commit 25ee09b

Please sign in to comment.