rename build_cstrs to build_model

SECQUOIA · Aug 3, 2024 · e212c4d · e212c4d
1 parent 35c8c8a
commit e212c4d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/gdplib/cstr/__init__.py b/gdplib/cstr/__init__.py
@@ -0,0 +1,3 @@
+from .gdp_reactor import build_model
+
+__all__ = ['build_model']
diff --git a/gdplib/cstr/gdp_reactor.py b/gdplib/cstr/gdp_reactor.py
@@ -17,7 +17,7 @@
 from pyomo.opt.base.solvers import SolverFactory
 
 
-def build_cstrs(NT: int = 5) -> pyo.ConcreteModel():
+def build_model(NT: int = 5) -> pyo.ConcreteModel():
     """
     Build the CSTR superstructure model of size NT.
     NT is the number of reactors in series.
@@ -915,7 +915,7 @@ def obj_rule(m):
 
 
 if __name__ == "__main__":
-    m = build_cstrs()
+    m = build_model()
     pyo.TransformationFactory("core.logical_to_linear").apply_to(m)
     pyo.TransformationFactory("gdp.bigm").apply_to(m)
     pyo.SolverFactory("gams").solve(