(
self,
drift,
*,
t0,
t1,
sampler_type,
num_steps,
atol,
rtol,
)
| 84 | """ODE solver class""" |
| 85 | |
| 86 | def __init__( |
| 87 | self, |
| 88 | drift, |
| 89 | *, |
| 90 | t0, |
| 91 | t1, |
| 92 | sampler_type, |
| 93 | num_steps, |
| 94 | atol, |
| 95 | rtol, |
| 96 | ): |
| 97 | # assert t0 < t1, "ODE sampler has to be in forward time", , comment it out, to make x2z ODE works |
| 98 | |
| 99 | self.drift = drift |
| 100 | self.t = th.linspace(t0, t1, num_steps) |
| 101 | self.atol = atol |
| 102 | self.rtol = rtol |
| 103 | self.sampler_type = sampler_type |
| 104 | |
| 105 | def sample(self, x, model, **model_kwargs): |
| 106 |
nothing calls this directly
no outgoing calls
no test coverage detected