fixed imports in examples

scaomath · Jul 20, 2024 · 8c64319 · 8c64319
1 parent b40b738
commit 8c64319
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/examples/ex2_FNO3d_train_normalized.ipynb b/examples/ex2_FNO3d_train_normalized.ipynb
@@ -32,12 +32,12 @@
     "\n",
     "import torch\n",
     "\n",
-    "from fno.pipeline import *\n",
-    "from fno.datasets import *\n",
-    "from fno.losses import SobolevLoss\n",
+    "from sfno.pipeline import *\n",
+    "from sfno.datasets import *\n",
+    "from sfno.losses import SobolevLoss\n",
     "\n",
-    "from fno.fno3d import FNO3d\n",
-    "from fno.visualizations import plot_contour_trajectory\n",
+    "from sfno.fno3d import FNO3d\n",
+    "from sfno.visualizations import plot_contour_trajectory\n",
     "from torch.utils.data import DataLoader\n",
     "\n",
     "from fno.utils import get_seed\n",

diff --git a/examples/ex2_SFNO_train.ipynb b/examples/ex2_SFNO_train.ipynb
@@ -12,8 +12,10 @@
     "import torch\n",
     "from sfno.utils import get_num_params, get_seed\n",
     "from sfno.pipeline import *\n",
-    "from sfno.visualizations import plot_contour_trajectory\n",
+    "from sfno.datasets import *\n",
+    "from sfno.losses import *\n",
     "from sfno.sfno import SFNO\n",
+    "from sfno.visualizations import plot_contour_trajectory\n",
     "from torch.utils.data import DataLoader\n",
     "get_seed(1127825, printout=False)\n",
     "\n",

diff --git a/examples/ex2_SFNO_train_fnodata.ipynb b/examples/ex2_SFNO_train_fnodata.ipynb
@@ -22,6 +22,8 @@
     "import torch\n",
     "from sfno.utils import get_num_params, get_seed\n",
     "from sfno.pipeline import *\n",
+    "from sfno.datasets import *\n",
+    "from sfno.losses import *\n",
     "from sfno.sfno import SFNO\n",
     "from sfno.visualizations import plot_contour_trajectory\n",
     "from torch.utils.data import DataLoader\n",