From 9cccceed2e9088862d24d2f090a39e85f4bcca02 Mon Sep 17 00:00:00 2001
From: Bart de Koning <bart.dekoning@deltares.nl>
Date: Tue, 23 Jul 2024 13:38:46 +0200
Subject: [PATCH] SPATIALCONTROLNODETYPES

---
 python/ribasim/ribasim/geometry/edge.py | 4 ++--
 ribasim_qgis/core/nodes.py              | 8 +++++++-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/python/ribasim/ribasim/geometry/edge.py b/python/ribasim/ribasim/geometry/edge.py
index 422d00b79..99f60d2aa 100644
--- 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
index ef3fe1e76..cbddfcef6 100644
--- 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]: