Fix wflow plotting

interTwin-eu · Sep 13, 2024 · 8afcb27 · 8afcb27
1 parent bdb2422
commit 8afcb27
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 10 deletions.
diff --git a/DT_flood/utils/plot_utils.py b/DT_flood/utils/plot_utils.py
@@ -20,6 +20,10 @@ def plot_wflow_model(wflow_model):
     wflow_model : hydromt_wflow.WFlowModel
         Wflow model instance from hydromt_wflow
     """    
+
+    if "wflow_dem" not in wflow_model.staticmaps:
+        wflow_model.read_staticmaps("staticmaps.nc")
+
     proj = ccrs.PlateCarree()
     gdf_bas = wflow_model.basins
     gdf_riv = wflow_model.rivers

diff --git a/Notebooks/SetupWFlow.ipynb b/Notebooks/SetupWFlow.ipynb
@@ -405,6 +405,25 @@
     "wf.write_config()"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "wf.write()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "wf_read = WflowModel(root=wf_root, mode=\"r\")\n",
+    "wf_read.read()"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 11,
@@ -435,16 +454,7 @@
     }
    ],
    "source": [
-    "plot_wflow_model(wf)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "wf.write()"
+    "plot_wflow_model(wf_read)"
    ]
   },
   {