Use np.ndarray in Constraints

2025-12-06 09:28:51 -06:00 · 2021-08-05 15:57:02 -05:00
parent 0c4b0ea81a
commit 0a32586bf8
5 changed files with 35 additions and 29 deletions
--- a/miplearn/solvers/pyomo/base.py
+++ b/miplearn/solvers/pyomo/base.py
@@ -236,11 +236,11 @@ class BasePyomoSolver(InternalSolver):

        return Constraints(
            names=_none_if_empty(names),
-            rhs=_none_if_empty(rhs),
+            rhs=_none_if_empty(np.array(rhs, dtype=float)),
            senses=_none_if_empty(senses),
            lhs=_none_if_empty(lhs),
-            slacks=_none_if_empty(slacks),
-            dual_values=_none_if_empty(dual_values),
+            slacks=_none_if_empty(np.array(slacks, dtype=float)),
+            dual_values=_none_if_empty(np.array(dual_values, dtype=float)),
        )

    @overrides