cleanup

bopen · Jul 19, 2023 · df518c6 · df518c6
1 parent c3daa04
commit df518c6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/notebooks/wp4/clima_and_bias_cordex_cmip6_regionalised.ipynb b/notebooks/wp4/clima_and_bias_cordex_cmip6_regionalised.ipynb
@@ -307,6 +307,8 @@
     "            da = da.assign_coords(height=ds[\"height\"] if \"height\" in ds else None)\n",
     "        else:\n",
     "            raise ValueError\n",
+    "    da = da.chunk(season=1, year=1)\n",
+    "    da.encoding[\"chunksizes\"] = tuple(map(max, da.chunks))\n",
     "    return da.to_dataset()"
    ]
   },
@@ -340,7 +342,7 @@
     "        },\n",
     "    )\n",
     "    .expand_dims(model=[\"ERA5\"])\n",
-    "    .chunk(chunks)\n",
+    "    .chunk(season=1, year=1)\n",
     ")\n",
     "\n",
     "datasets = []\n",
@@ -363,7 +365,7 @@
     "        },\n",
     "    )\n",
     "    datasets.append(ds.expand_dims(model=[model]))\n",
-    "ds_sim = xr.concat(datasets, \"model\").chunk(chunks)"
+    "ds_sim = xr.concat(datasets, \"model\").chunk(season=1, year=1)"
    ]
   },
   {