Define build_components as variable in flowsheet_two_stage

watertap-org · Apr 5, 2023 · f4586a4 · f4586a4
1 parent 097b2c7
commit f4586a4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/watertap/examples/flowsheets/full_treatment_train/analysis/flowsheet_two_stage.py b/watertap/examples/flowsheets/full_treatment_train/analysis/flowsheet_two_stage.py
@@ -29,6 +29,7 @@
 import watertap.examples.flowsheets.full_treatment_train.analysis.flowsheet_single_stage as single_stage
 from watertap.examples.flowsheets.full_treatment_train.analysis.flowsheet_single_stage import (
     build,
+    build_components as build_components_single_stage,
     scale,
     initialize,
     report,
@@ -45,6 +46,8 @@
     "RO_level": "detailed",
 }
 
+build_components = build_components_single_stage
+
 
 def set_optimization_components(m, system_recovery, **kwargs):
     single_stage.set_optimization_components(m, system_recovery, **kwargs)