Convert LPSolveStats into dataclass

2025-12-06 09:28:51 -06:00 · 2021-04-11 08:41:50 -05:00
parent 6afdf2ed55
commit 945f6a091c
7 changed files with 24 additions and 28 deletions
--- a/miplearn/solvers/pyomo/base.py
+++ b/miplearn/solvers/pyomo/base.py
@@ -87,10 +87,10 @@ class BasePyomoSolver(InternalSolver):
        if not self.is_infeasible():
            opt_value = results["Problem"][0]["Lower bound"]
            self._has_lp_solution = True
-        return {
-            "LP value": opt_value,
-            "LP log": streams[0].getvalue(),
-        }
+        return LPSolveStats(
+            lp_value=opt_value,
+            lp_log=streams[0].getvalue(),
+        )

    def _restore_integrality(self) -> None:
        for var in self._bin_vars: