fix cordex

bopen · Feb 22, 2024 · 5d7335f · 5d7335f
1 parent b7b1fae
commit 5d7335f
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/notebooks/wp4/extreme_temperature_indices.ipynb b/notebooks/wp4/extreme_temperature_indices.ipynb
@@ -304,13 +304,14 @@
     "    xr.save_mfdataset(datasets, paths)\n",
     "\n",
     "    ds = xr.open_mfdataset(paths)\n",
+    "    print(ds)\n",
     "    (da,) = ds.data_vars.values()\n",
     "    mask = da.notnull().any(\"time\")\n",
     "\n",
     "    in_files = f\"{tmpdir}/rechunked.zarr\"\n",
-    "    ds.chunk(\n",
-    "        {dim: -1 if dim == \"time\" else \"auto\" for dim in set(ds.dims) & set(ds.cf.axes)}\n",
-    "    ).to_zarr(in_files)\n",
+    "    ds.chunk({dim: -1 if dim == \"time\" else \"auto\" for dim in da.dims}).to_zarr(\n",
+    "        in_files\n",
+    "    )\n",
     "\n",
     "    datasets = [\n",
     "        icclim.index(\n",
@@ -321,6 +322,7 @@
     "        )\n",
     "        for index_name in index_names\n",
     "    ]\n",
+    "\n",
     "    return xr.merge(datasets).where(mask).drop_dims(\"bounds\")\n",
     "\n",
     "\n",
@@ -356,6 +358,10 @@
     "    resample,\n",
     "    **regrid_kwargs,\n",
     "):\n",
+    "    if \"rotated_pole\" in ds:\n",
+    "        ds = ds.drop_vars(\"rotated_pole\")\n",
+    "\n",
+    "    # Original bounds for conservative interpolation\n",
     "    if regrid_kwargs.get(\"method\") == \"conservative\":\n",
     "        bounds = [\n",
     "            ds.cf.get_bounds(coord).reset_coords(drop=True)\n",