Also update discrete control table.

Deltares · Sep 10, 2024 · f7c95c8 · f7c95c8 · DanielTollenaar · Sep 11, 2024
1 parent 5613858
commit f7c95c8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/ribasim_nl/ribasim_nl/reset_index.py b/src/ribasim_nl/ribasim_nl/reset_index.py
@@ -34,6 +34,10 @@ def reset_index(model: Model, node_start=1):
                     elif table.df.index.name == "node_id":
                         table.df.index = table.df.reset_index("node_id")["node_id"].apply(lambda x: index.loc[x])
                         # table.df.index.name = "node_id"
+
+                    if "listen_node_id" in table.df.columns:
+                        table.df.loc[:, "listen_node_id"] = table.df["listen_node_id"].apply(lambda x: index[x])
+
     return model