minor changes on comments and notebook

DiffEqML · Nov 18, 2023 · 880e855 · 880e855
1 parent f9b6afd
commit 880e855
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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": [