updates on flow modeling

Author: mlmamud

examples/example_7_flow_2d_pinn_on_XCT.ipynb

@@ -36,10 +36,19 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 56,
+   "execution_count": 58,
    "id": "1a7cff2d-f2ab-41a5-a5cf-402c6e32c54a",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "The autoreload extension is already loaded. To reload it, use:\n",
+      "  %reload_ext autoreload\n"
+     ]
+    }
+   ],
    "source": [
     "import os\n",
     "import sys\n",
@@ -49,8 +58,8 @@
     "import jax.numpy as jnp\n",
     "from jax import random, jit, grad, vmap\n",
     "import optax\n",
-    "# %load_ext autoreload\n",
-    "# %autoreload 2\n",
+    "%load_ext autoreload\n",
+    "%autoreload 2\n",
     "\n",
     "# import functions\n",
     "current_directory = os.getcwd()                             # current directory\n",