More fixes

Deltares · Aug 1, 2024 · 4ed875b · 4ed875b
1 parent 69de52e
commit 4ed875b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 14 deletions.
diff --git a/src/peilbeheerst_model/Parametrize/Rijnland_parametrize.ipynb b/src/peilbeheerst_model/Parametrize/Rijnland_parametrize.ipynb
@@ -16,6 +16,7 @@
     "import peilbeheerst_model.ribasim_parametrization as ribasim_param\n",
     "import ribasim\n",
     "import ribasim.nodes\n",
+    "from peilbeheerst_model.add_storage_basins import AddStorageBasin\n",
     "\n",
     "warnings.filterwarnings(\"ignore\")"
    ]

diff --git a/src/peilbeheerst_model/peilbeheerst_model/preprocess_data/HHSK.ipynb b/src/peilbeheerst_model/peilbeheerst_model/preprocess_data/HHSK.ipynb
@@ -26,7 +26,7 @@
    },
    "outputs": [],
    "source": [
-    "from general_functions import read_gpkg_layers, show_layers_and_columns, store_data"
+    "from general_functions import burn_in_peilgebieden, read_gpkg_layers, show_layers_and_columns, store_data"
    ]
   },
   {
@@ -376,17 +376,21 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# streefpeilen_PG_v = pd.merge(left = HHSK['peilgebiedvigerend'],\n",
-    "#                              right = HHSK['streefpeil'],\n",
-    "#                              left_on = 'globalid',\n",
-    "#                              right_on = 'peilgebiedvigerendid',\n",
-    "#                              suffixes = ('', '_streefpeil'))[['code', 'nen3610id', 'globalid', 'waterhoogte', 'geometry']]\n",
+    "streefpeilen_PG_v = pd.merge(\n",
+    "    left=HHSK[\"peilgebiedvigerend\"],\n",
+    "    right=HHSK[\"streefpeil\"],\n",
+    "    left_on=\"globalid\",\n",
+    "    right_on=\"peilgebiedvigerendid\",\n",
+    "    suffixes=(\"\", \"_streefpeil\"),\n",
+    ")[[\"code\", \"nen3610id\", \"globalid\", \"waterhoogte\", \"geometry\"]]\n",
     "\n",
-    "# streefpeilen_PG_a = pd.merge(left = HHSK['peilafwijkinggebied'],\n",
-    "#                              right = HHSK['streefpeil'],\n",
-    "#                              left_on = 'globalid',\n",
-    "#                              right_on = 'peilafwijkinggebiedid',\n",
-    "#                              suffixes = ('', '_streefpeil'))[['code', 'nen3610id', 'globalid', 'waterhoogte', 'geometry']]\n",
+    "streefpeilen_PG_a = pd.merge(\n",
+    "    left=HHSK[\"peilafwijkinggebied\"],\n",
+    "    right=HHSK[\"streefpeil\"],\n",
+    "    left_on=\"globalid\",\n",
+    "    right_on=\"peilafwijkinggebiedid\",\n",
+    "    suffixes=(\"\", \"_streefpeil\"),\n",
+    ")[[\"code\", \"nen3610id\", \"globalid\", \"waterhoogte\", \"geometry\"]]\n",
     "\n",
     "# fig, ax = plt.subplots()\n",
     "# streefpeilen_PG_v.plot(ax = ax, color='cornflowerblue')\n",
@@ -411,9 +415,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# peilgebied = burn_in_peilgebieden(base_layer = streefpeilen_PG_v,\n",
-    "#                                   overlay_layer = streefpeilen_PG_a,\n",
-    "#                                   plot = True)"
+    "peilgebied = burn_in_peilgebieden(base_layer=streefpeilen_PG_v, overlay_layer=streefpeilen_PG_a, plot=True)"
    ]
   },
   {