(self, x, mean_x, t, model, **model_kwargs)
| 219 | return xhat + 0.5 * self.dt * (K1 + K2), xhat # at last time point we do not perform the heun step |
| 220 | |
| 221 | def __call__(self, x, mean_x, t, model, **model_kwargs): |
| 222 | return self.sampler(x, mean_x, t, model, **model_kwargs) |
| 223 | |
| 224 | |
| 225 | class FlowSDE: |
nothing calls this directly
no outgoing calls
no test coverage detected