fixed imports in examples

Author: scaomath

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",

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",

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",