cleanup

bopen · Feb 7, 2024 · 76021f4 · 76021f4
1 parent 6bde2ff
commit 76021f4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/notebooks/wp3/WIP-hit_rate.ipynb b/notebooks/wp3/WIP-hit_rate.ipynb
@@ -221,7 +221,8 @@
     "    # Mask region\n",
     "    mask = regionmask.defined_regions.srex.mask(ds)\n",
     "    index = regionmask.defined_regions.srex.map_keys(region)\n",
-    "    ds = ds.where((mask == index).compute(), drop=True).expand_dims(region=[region])\n",
+    "    ds = ds.where((mask == index).compute(), drop=True)\n",
+    "    ds = ds.expand_dims(region=[region])\n",
     "\n",
     "    # Spatial mean\n",
     "    ds = diagnostics.spatial_weighted_mean(ds)\n",