fix seasonal

Author: malmans2

notebooks/wp3/seasonal_forecast_vs_oras5.ipynb

@@ -26,6 +26,7 @@
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import pandas as pd\n",
+    "import pooch\n",
     "import xarray as xr\n",
     "from c3s_eqc_automatic_quality_control import diagnostics, download, plot\n",
     "\n",
@@ -123,13 +124,13 @@
    "source": [
     "def add_bounds(ds):\n",
     "    # From https://github.com/COSIMA/ocean-regrid/blob/master/nemo_grid.py\n",
-    "    dg = xr.open_dataset(\n",
-    "        (\n",
-    "            \"https://icdc.cen.uni-hamburg.de/thredds/dodsC/ftpthredds/\"\n",
-    "            \"EASYInit/oras5/ORCA025/mesh/mesh_mask.nc\"\n",
-    "        ),\n",
-    "        chunks={},\n",
-    "    ).isel(t=0, z=0)\n",
+    "    url = (\n",
+    "        \"https://icdc.cen.uni-hamburg.de/thredds/fileServer/ftpthredds/\"\n",
+    "        \"EASYInit/oras5/ORCA025/mesh/mesh_mask.nc\"\n",
+    "    )\n",
+    "    known_hash = \"ba88418bc8055972b675ceedbd4345b3827f0ede8cef04d53c97b66a5c7a46a7\"\n",
+    "    fname = pooch.retrieve(url, known_hash)\n",
+    "    dg = xr.open_dataset(fname, chunks={}).isel(t=0, z=0)\n",
     "\n",
     "    # These are the top righ-hand corner of t cells.\n",
     "    glamf = dg.glamf\n",
@@ -328,13 +329,17 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "month_order = [10, 11, 12, 1, 2]\n",
     "datasets = {\"Reanalysis\": ds_reanalysis, \"Seasonal Forecast\": ds_seasonal}\n",
     "for product, ds in datasets.items():\n",
     "    (da,) = ds.data_vars.values()\n",
+    "    da = da.sel(month=month_order)\n",
     "    if \"realization\" in da.dims:\n",
     "        da = da.mean(\"realization\", keep_attrs=True)\n",
     "\n",
-    "    _ = plot.projected_map(da.mean(\"year\", keep_attrs=True), col=\"month\", col_wrap=3)\n",
+    "    _ = plot.projected_map(\n",
+    "        da.mean(\"year\", keep_attrs=True), col=\"month\", col_wrap=3, robust=True\n",
+    "    )\n",
     "    plt.suptitle(f\"{product = }\")\n",
     "    plt.show()\n",
     "\n",