SPATIALCONTROLNODETYPES

Deltares · Jul 23, 2024 · 9ccccee · 9ccccee
1 parent af48e2c
commit 9ccccee
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/python/ribasim/ribasim/geometry/edge.py b/python/ribasim/ribasim/geometry/edge.py
@@ -17,10 +17,10 @@
 __all__ = ("EdgeTable",)
 
 SPATIALCONTROLNODETYPES = {
-    "LevelDemand",
-    "FlowDemand",
     "ContinuousControl",
     "DiscreteControl",
+    "FlowDemand",
+    "LevelDemand",
     "PidControl",
 }
 

diff --git a/ribasim_qgis/core/nodes.py b/ribasim_qgis/core/nodes.py
@@ -922,7 +922,13 @@ def attributes(cls) -> list[QgsField]:
     cls.nodetype() for cls in Input.__subclasses__() if not cls.is_spatial()
 }
 EDGETYPES = {"flow", "control"}
-SPATIALCONTROLNODETYPES = {"LevelDemand", "FlowDemand", "DiscreteControl", "PidControl"}
+SPATIALCONTROLNODETYPES = {
+    "ContinuousControl",
+    "DiscreteControl",
+    "FlowDemand",
+    "LevelDemand",
+    "PidControl",
+}
 
 
 def load_nodes_from_geopackage(path: Path) -> dict[str, Input]: