diff --git a/test/test_sdeint.py b/test/test_sdeint.py index f9811ae..1dc3ed7 100644 --- a/test/test_sdeint.py +++ b/test/test_sdeint.py @@ -63,7 +63,6 @@ def g(self, t, x): assert_almost_equal(xs_true[0][-1], xs_torchdyn[-1], decimal=2) -# todo : need to improve sdeint for stratonovich @pytest.mark.parametrize("solver", ["eulerHeun", "milstein_stratonovich"]) def test_geo_brownian_stratonovich(solver): torch.manual_seed(0) diff --git a/torchdyn/numerics/solvers/sde.py b/torchdyn/numerics/solvers/sde.py index e598a3d..328f440 100644 --- a/torchdyn/numerics/solvers/sde.py +++ b/torchdyn/numerics/solvers/sde.py @@ -16,7 +16,6 @@ def _check_types(sde, solver_sde_type, solver_noise_type): ) -# todo : Should I make additional SDEqSolver?? so many duplicates here class EulerMaruyama(DiffEqSolver): def __init__(self, sde: SDEFunc, bm: BaseBrownian, dtype=torch.float32): super().__init__(order=1) diff --git a/tutorials/module1-neuralde/m1e_neural_sde_cookbook.ipynb b/tutorials/module1-neuralde/m1e_neural_sde_cookbook.ipynb index e417e88..e0da12e 100644 --- a/tutorials/module1-neuralde/m1e_neural_sde_cookbook.ipynb +++ b/tutorials/module1-neuralde/m1e_neural_sde_cookbook.ipynb @@ -1,8 +1,17 @@ { "cells": [ + { + "cell_type": "markdown", + "id": "48f053c3", + "metadata": {}, + "source": [ + "# Neural SDEs Cookbook\n", + "In this notebook, we explore the Neural SDE module in torchdyn. " + ] + }, { "cell_type": "code", - "execution_count": 30, + "execution_count": null, "id": "579b6a9a", "metadata": {}, "outputs": [