Minor clean-ups

Author: erikvansebille

docs/user_guide/examples/tutorial_nestedgrids.ipynb

@@ -40,7 +40,9 @@
    "id": "2",
    "metadata": {},
    "source": [
-    "## Setting up the individual Grids and Fields"
+    "## Setting up the individual Grids and Fields\n",
+    "\n",
+    "Before we start, define a helper function to quickly set up a nested dataset"
    ]
   },
   {
@@ -54,12 +56,9 @@
     "grid_polygons = []\n",
     "\n",
     "\n",
-    "def setup_nestedFields(polygon):\n",
+    "def setup_nested_ds(polygon):\n",
     "    xdim, ydim = 20, 20\n",
-    "\n",
-    "    ds = simple_UV_dataset(dims=(1, 1, ydim, xdim), mesh=\"spherical\").isel(\n",
-    "        time=0, depth=0\n",
-    "    )\n",
+    "    ds = simple_UV_dataset(dims=(1, 1, ydim, xdim), mesh=\"spherical\")\n",
     "    ds[\"lon\"][:] = np.linspace(polygon[:, 0].min(), polygon[:, 0].max(), xdim)\n",
     "    ds[\"lat\"][:] = np.linspace(polygon[:, 1].min(), polygon[:, 1].max(), ydim)\n",
     "    return ds"
@@ -83,7 +82,7 @@
     "polygon = np.array([(10, 15), (25, 10), (25, 25), (10, 35)])\n",
     "grid_polygons.append(polygon)\n",
     "\n",
-    "ds_in.append(setup_nestedFields(polygon))\n",
+    "ds_in.append(setup_nested_ds(polygon))\n",
     "ds_in[-1][\"U\"][:] = 1.0\n",
     "ds_in[-1][\"V\"][:] = 0.0"
    ]
@@ -106,7 +105,7 @@
     "polygon = np.array([(0, -5), (35, 0), (35, 25), (0, 20)])\n",
     "grid_polygons.append(polygon)\n",
     "\n",
-    "ds_in.append(setup_nestedFields(polygon))\n",
+    "ds_in.append(setup_nested_ds(polygon))\n",
     "ds_in[-1][\"U\"][:] = 1.0\n",
     "ds_in[-1][\"V\"][:] = 1.0"
    ]
@@ -129,7 +128,7 @@
     "polygon = np.array([(-10, -20), (60, -20), (60, 40), (-10, 40)])\n",
     "grid_polygons.append(polygon)\n",
     "\n",
-    "ds_in.append(setup_nestedFields(polygon))\n",
+    "ds_in.append(setup_nested_ds(polygon))\n",
     "lon_g, lat_g = np.meshgrid(ds_in[-1][\"lon\"], ds_in[-1][\"lat\"])\n",
     "ds_in[-1][\"U\"][:] = 1.0\n",
     "ds_in[-1][\"V\"][:] = np.cos(lon_g / 5 * np.pi / 2)"
@@ -156,7 +155,7 @@
     "fig, ax = plt.subplots()\n",
     "\n",
     "for i in range(n_grids):\n",
-    "    ds = ds_in[i]\n",
+    "    ds = ds_in[i].isel(time=0, depth=0)\n",
     "    ax.quiver(ds.lon, ds.lat, ds.U, ds.V, color=f\"C{i}\")\n",
     "    poly = grid_polygons[i]\n",
     "    ax.plot(\n",