fix gw tot discharge

nens · Dec 9, 2024 · 197d78c · 197d78c
1 parent 2cd5a11
commit 197d78c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/threedigrid_builder/grid/boundary_conditions.py b/threedigrid_builder/grid/boundary_conditions.py
@@ -18,7 +18,11 @@
 
 
 GROUNDWATER_BOUNDARY_TYPES = frozenset(
-    {BoundaryType.GROUNDWATERLEVEL, BoundaryType.GROUNDWATERDISCHARGE}
+    {
+        BoundaryType.GROUNDWATERLEVEL,
+        BoundaryType.GROUNDWATERDISCHARGE,
+        BoundaryType.GROUNDWATER_TOTAL_DISCHARGE_2D,
+    }
 )
 
 
@@ -270,6 +274,7 @@ def boundary_type_is_groundwater(type: BoundaryType) -> bool:
         return type in {
             BoundaryType.GROUNDWATERLEVEL,
             BoundaryType.GROUNDWATERDISCHARGE,
+            BoundaryType.GROUNDWATER_TOTAL_DISCHARGE_2D,
         }
 
     def check_edge_coord(