From 3c9eccdfab49ead2ed8c4490ba577dae76dcb2cd Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <pierre-antoine.rouby@inrae.fr>
Date: Tue, 23 Jan 2024 11:30:34 +0100
Subject: [PATCH] Solver: Mage: Fix <list>.node error.

---
 src/Solver/Mage.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/Solver/Mage.py b/src/Solver/Mage.py
index acd94fad..2189e7bf 100644
--- a/src/Solver/Mage.py
+++ b/src/Solver/Mage.py
@@ -234,7 +234,7 @@ class Mage(CommandLineSolver):
     def _export_BC(self, t, bounds, repertory, qlog, name="0"):
         files = []
 
-        if len(bounds) == 0 or bounds.node is None:
+        if len(bounds) == 0:
             return files
 
         if qlog is not None:
@@ -244,6 +244,9 @@ class Mage(CommandLineSolver):
             files.append(f"{name}.{t}")
 
             for bound in bounds:
+                if bound.node is None:
+                    continue
+
                 name = f"{bound.node.id:3}".replace(" ", "x")
                 f.write(f"* {bound.node.name} ({name}) {bound.bctype}\n")
                 f.write(f"${name}\n")
@@ -419,7 +422,7 @@ class Mage(CommandLineSolver):
         files = []
 
         reservoirs = study.river.reservoir.lst
-        if len(reservoirs) == 0 or reservoirs.node is None:
+        if len(reservoirs) == 0:
             return files
 
         if qlog is not None:
@@ -429,6 +432,9 @@ class Mage(CommandLineSolver):
             files.append(f"{name}.CAS")
 
             for reservoir in reservoirs:
+                if reservoir.node is None:
+                    continue
+
                 reservoir.sort()
                 node = reservoir.node
                 name = f"{node.id:3}".replace(" ", "x")
-- 
GitLab