Merge branch 'main' into 86-sensitivity-analysis-sampling

ImperialCollegeLondon · Mar 20, 2024 · 9046178 · 9046178
2 parents e4a38e5 + 6f216e3
commit 9046178
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/swmmanywhere/graph_utilities.py b/swmmanywhere/graph_utilities.py
@@ -177,10 +177,11 @@ def iterate_graphfcns(G: nx.Graph,
     not_exists = [g for g in graphfcn_list if g not in graphfcns]
     if not_exists:
         raise ValueError(f"Graphfcns are not registered:\n{', '.join(not_exists)}")
+    verbose = os.getenv("SWMMANYWHERE_VERBOSE", "false").lower() == "true"
     for function in graphfcn_list:
         G = graphfcns[function](G, addresses = addresses, **params)
         logger.info(f"graphfcn: {function} completed.")
-        if os.getenv("SWMMANYWHERE_VERBOSE", "false").lower() == "true":
+        if verbose:
             save_graph(G, addresses.model / f"{function}_graph.json")
     return G