Doc fix: docs/user_guide/examples/tutorial_dt_integrators.ipynb

Author: VeckoTheGecko

docs/user_guide/examples/tutorial_dt_integrators.ipynb

@@ -53,6 +53,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from pathlib import Path\n",
+    "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import pandas as pd\n",
@@ -67,6 +69,9 @@
     ")\n",
     "ds_fields.load()  # load the dataset into memory\n",
     "\n",
+    "OUTPUT_FOLDER = Path(\"output\")\n",
+    "OUTPUT_FOLDER.mkdir(exist_ok=True)\n",
+    "\n",
     "# Convert to SGRID-compliant dataset and create FieldSet\n",
     "fields = {\"U\": ds_fields[\"uo\"], \"V\": ds_fields[\"vo\"]}\n",
     "ds_fset = parcels.convert.copernicusmarine_to_sgrid(fields=fields)\n",
@@ -239,7 +244,8 @@
     "    outputdt = dt\n",
     "\n",
     "    pfile = parcels.ParticleFile(\n",
-    "        path=f\"output/AdvectionRK2_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\",\n",
+    "        path=OUTPUT_FOLDER\n",
+    "        / f\"AdvectionRK2_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\",\n",
     "        outputdt=outputdt,\n",
     "    )\n",
     "\n",
@@ -291,7 +297,7 @@
     "temperature = ds_fields.isel(time=0, depth=0).thetao.plot(cmap=\"Greys\")\n",
     "for j, dt in enumerate(dt_choices):\n",
     "    df = parcels.read_particlefile(\n",
-    "        f\"output/AdvectionRK2_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\"\n",
+    "        OUTPUT_FOLDER / f\"AdvectionRK2_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\"\n",
     "    )\n",
     "    for i, traj in enumerate(df.partition_by(\"particle_id\", maintain_order=True)):\n",
     "        ax.plot(\n",
@@ -376,10 +382,10 @@
     "axs[1].set_ylim(0, 50)\n",
     "\n",
     "# set 5 minute dt as benchmark\n",
-    "df_5min = parcels.read_particlefile(\"output/AdvectionRK2_dt_300s.parquet\")\n",
+    "df_5min = parcels.read_particlefile(OUTPUT_FOLDER / \"AdvectionRK2_dt_300s.parquet\")\n",
     "for i, dt in enumerate(dt_choices[:-1]):\n",
     "    df = parcels.read_particlefile(\n",
-    "        f\"output/AdvectionRK2_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\"\n",
+    "        OUTPUT_FOLDER / f\"AdvectionRK2_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\"\n",
     "    )\n",
     "\n",
     "    # subset 5 minute data to match dt\n",
@@ -581,7 +587,8 @@
     "        outputdt = dt\n",
     "\n",
     "        pfile = parcels.ParticleFile(\n",
-    "            path=f\"output/KernelCompare_{advection_scheme.__name__}_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\",\n",
+    "            path=OUTPUT_FOLDER\n",
+    "            / f\"KernelCompare_{advection_scheme.__name__}_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\",\n",
     "            outputdt=outputdt,\n",
     "        )\n",
     "\n",
@@ -620,7 +627,8 @@
     "    axs[m, n].set_xlabel(\"Longitude\")\n",
     "    for j, advection_scheme in enumerate(advection_schemes):\n",
     "        df = parcels.read_particlefile(\n",
-    "            f\"output/KernelCompare_{advection_scheme.__name__}_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\"\n",
+    "            OUTPUT_FOLDER\n",
+    "            / f\"KernelCompare_{advection_scheme.__name__}_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\"\n",
     "        )\n",
     "        for i, traj in enumerate(df.partition_by(\"particle_id\", maintain_order=True)):\n",
     "            axs[m, n].plot(\n",
@@ -670,11 +678,13 @@
     "    axs[m, n].set_yscale(\"log\")\n",
     "    axs[m, n].set_ylim(1e-4, 1e1)\n",
     "    df_RK4 = parcels.read_particlefile(\n",
-    "        f\"output/KernelCompare_AdvectionRK4_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\"\n",
+    "        OUTPUT_FOLDER\n",
+    "        / f\"KernelCompare_AdvectionRK4_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\"\n",
     "    )\n",
     "    for j, advection_scheme in enumerate(advection_schemes[:-1]):\n",
     "        df = parcels.read_particlefile(\n",
-    "            f\"output/KernelCompare_{advection_scheme.__name__}_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\"\n",
+    "            OUTPUT_FOLDER\n",
+    "            / f\"KernelCompare_{advection_scheme.__name__}_dt_{int(dt / np.timedelta64(1, 's'))}s.parquet\"\n",
     "        )\n",
     "\n",
     "        dist = dist_km(df[\"lon\"], df_RK4[\"lon\"], df[\"lat\"], df_RK4[\"lat\"])\n",