fix import and scripts typos (Fixes #1 )

* fix __init__.py in sdetools for missing reference to VPSDE. * rename run_mnist.sh to train_minist.sh.
giulio98 · Dec 18, 2023 · 5e22f9f · 5e22f9f
1 parent e6daef7
commit 5e22f9f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/scripts/maml/run_mnist.sh → scripts/maml/train_mnist.sh b/scripts/maml/run_mnist.sh → scripts/maml/train_mnist.sh
diff --git a/src/functional_diffusion_processes/sdetools/__init__.py b/src/functional_diffusion_processes/sdetools/__init__.py
@@ -1,5 +1,4 @@
 from .base_sde import SDE
 from .heat_subvp_sde import HeatSubVPSDE
-from .vp_sde import VPSDE
 
-__all__ = ["SDE", "HeatSubVPSDE", "VPSDE"]
+__all__ = ["SDE", "HeatSubVPSDE"]