fix(barends): add conditionals in scenario function (#229)

MODFLOW-USGS · Oct 3, 2024 · 4748ef0 · 4748ef0
1 parent c5b10d3
commit 4748ef0
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/scripts/ex-gwe-barends.py b/scripts/ex-gwe-barends.py
@@ -606,14 +606,15 @@ def scenario(idx, silent=False):
  sim_gwf = build_mf6_flow_model()
  sim_gwe = build_mf6_heat_model()
 
- if sim_gwf is not None and sim_gwe is not None:
+ if write and (sim_gwf is not None and sim_gwe is not None):
  write_mf6_models(sim_gwf, sim_gwe, silent=silent)
 
- success = run_model(sim_gwf, silent=silent)
- if success:
- success = run_model(sim_gwe, silent=silent)
+ if run:
+ success = run_model(sim_gwf, silent=silent)
+ if success:
+ success = run_model(sim_gwe, silent=silent)
 
- if success:
+ if plot and success:
  plot_thermal_bleeding(sim_gwe)