Quarto output

Author: MartinSchobben

notebooks/courses/microwave-remote-sensing/unit_03/09_in_class_exercise.ipynb

@@ -30,15 +30,17 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "import contextlib\n",
+    "import os\n",
+    "\n",
     "import cmcrameri as cmc  # noqa: F401\n",
     "import intake\n",
     "import matplotlib.patches as patches\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import seaborn as sns\n",
     "import snaphu\n",
-    "import xarray as xr\n",
-    "from IPython.display import clear_output"
+    "import xarray as xr"
    ]
   },
   {
@@ -120,6 +122,24 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "@contextlib.contextmanager\n",
+    "def suppress_output():\n",
+    "    with open(os.devnull, \"w\") as devnull:\n",
+    "        old_stdout = os.dup(1)\n",
+    "        old_stderr = os.dup(2)\n",
+    "\n",
+    "        os.dup2(devnull.fileno(), 1)\n",
+    "        os.dup2(devnull.fileno(), 2)\n",
+    "\n",
+    "        try:\n",
+    "            yield\n",
+    "        finally:\n",
+    "            os.dup2(old_stdout, 1)\n",
+    "            os.dup2(old_stderr, 2)\n",
+    "            os.close(old_stdout)\n",
+    "            os.close(old_stderr)\n",
+    "\n",
+    "\n",
     "def unwrap_array(\n",
     "    data: xr.DataArray,\n",
     "    complex_var: str = \"cmplx\",\n",
@@ -176,18 +196,16 @@
     "        coherence = np.ones_like(data_arr.real)\n",
     "\n",
     "    # Unwrap the phase (already in complex form)\n",
-    "    unw, _ = snaphu.unwrap(\n",
-    "        data_arr,\n",
-    "        coherence,\n",
-    "        nlooks=nlooks,\n",
-    "        cost=cost,\n",
-    "        init=init,\n",
-    "        mask=mask,\n",
-    "        **kwargs,\n",
-    "    )\n",
-    "\n",
-    "    # Clear the output to avoid printing the snaphu output\n",
-    "    clear_output()\n",
+    "    with suppress_output():\n",
+    "        unw, _ = snaphu.unwrap(\n",
+    "            data_arr,\n",
+    "            coherence,\n",
+    "            nlooks=nlooks,\n",
+    "            cost=cost,\n",
+    "            init=init,\n",
+    "            mask=mask,\n",
+    "            **kwargs,\n",
+    "        )\n",
     "\n",
     "    # Build xarray DataArray with the unwrapped phase\n",
     "    # unw_da = xr.DataArray(unw, coords=data.coords, dims=data.dims)\n",