Skip to content

Commit 880e855

Browse files
committed
minor changes on comments and notebook
1 parent f9b6afd commit 880e855

File tree

3 files changed

+10
-3
lines changed

3 files changed

+10
-3
lines changed

test/test_sdeint.py

-1
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,6 @@ def g(self, t, x):
6363
assert_almost_equal(xs_true[0][-1], xs_torchdyn[-1], decimal=2)
6464

6565

66-
# todo : need to improve sdeint for stratonovich
6766
@pytest.mark.parametrize("solver", ["eulerHeun", "milstein_stratonovich"])
6867
def test_geo_brownian_stratonovich(solver):
6968
torch.manual_seed(0)

torchdyn/numerics/solvers/sde.py

-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ def _check_types(sde, solver_sde_type, solver_noise_type):
1616
)
1717

1818

19-
# todo : Should I make additional SDEqSolver?? so many duplicates here
2019
class EulerMaruyama(DiffEqSolver):
2120
def __init__(self, sde: SDEFunc, bm: BaseBrownian, dtype=torch.float32):
2221
super().__init__(order=1)

tutorials/module1-neuralde/m1e_neural_sde_cookbook.ipynb

+10-1
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,17 @@
11
{
22
"cells": [
3+
{
4+
"cell_type": "markdown",
5+
"id": "48f053c3",
6+
"metadata": {},
7+
"source": [
8+
"# Neural SDEs Cookbook\n",
9+
"In this notebook, we explore the Neural SDE module in torchdyn. "
10+
]
11+
},
312
{
413
"cell_type": "code",
5-
"execution_count": 30,
14+
"execution_count": null,
615
"id": "579b6a9a",
716
"metadata": {},
817
"outputs": [

0 commit comments

Comments
 (0)