search

pnkraemer / probdiffeq

Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and custom information operators. Compatible with the broader JAX scientific computing ecosystem.
https://pnkraemer.github.io/probdiffeq/
MIT License
30 stars 2 forks source link

Unnecessary dependency on Diffeqzoo & Diffrax in v0.4.0 #762

Closed adam-hartshorne closed 2 months ago

adam-hartshorne commented 2 months ago

With v0.4.0 and the rejigging of imports, you have now made a dependence on diffeqzoo and diffrax.

taylor -> ode -> diffeqzoo.

from probdiffeq import ivpsolve, ivpsolvers, stats, taylor

taylor.py from probdiffeq.backend import control_flow, functools, itertools, ode, tree_util

ode.py

import diffeqzoo.ivps
import diffrax
pnkraemer commented 2 months ago

Hm. The dependence has always been there. You might not have encountered it because you haven't used the Runge-Kutta starter. That said, it should not exist. I will take care of this.

adam-hartshorne commented 2 months ago

No I definitely using autodiff.taylor_mode_scan before.

pnkraemer commented 2 months ago

No I definitely using autodiff.taylor_mode_scan before.

I thought so :)

The issue has been resolved, and a new release has been published. Upgrade your probdiffeq version, and you should be good to go. Thanks for reporting.

adam-hartshorne commented 2 months ago

Thanks.