From e19c3229bd9e509d2279138fa16f58fabecb2131 Mon Sep 17 00:00:00 2001 From: Pierre-Antoine Rouby Date: Fri, 5 Jun 2026 17:01:22 +0200 Subject: [PATCH] Solver: Mage: HS: Minor change. --- src/Solver/Mage.py | 23 ++++++++++------------- 1 file changed, 10 insertions(+), 13 deletions(-) diff --git a/src/Solver/Mage.py b/src/Solver/Mage.py index 798fad19..fef74d27 100644 --- a/src/Solver/Mage.py +++ b/src/Solver/Mage.py @@ -546,6 +546,15 @@ class Mage(CommandLineSolver): return files + def skip_hs(self, hs): + return ( + hs.input_reach is None + or not hs.input_reach.is_enable() + or not hs.enabled + or hs.input_section is None + or hs.is_deleted() + ) + def _export_SIN(self, study, repertory, qlog, name="0"): files = [] @@ -571,19 +580,7 @@ class Mage(CommandLineSolver): files.append(f"{name}.SIN") for hs in hydraulic_structures: - if hs.input_reach is None: - continue - - if not hs.input_reach.is_enable(): - continue - - if not hs.enabled: - continue - - if hs.input_section is None: - continue - - if hs.is_deleted(): + if self.skip_hs(hs): continue f.write(