By default, diffrax.diffeqsolve uses the argument throw=True. In some cases this may cause field_line_integrate to terminate prematurely with an error. It might be better to set throw=False, and then check what data can still be safely returned. This would allow the user to plot partial field lines trajectories up until the integration failed.