ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 09:28:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

GurobiSolver: Accept integer variables, as long as bounds=(0,1)

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-13 10:39:36 -05:00
parent c4a6665825
commit 25affca3ec
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
miplearn/solvers/gurobi.py
View File

@@ -476,12 +476,23 @@ class GurobiSolver(InternalSolver):
f"Unique variable names are currently required."
)
self._varname_to_var[var.varName] = var
assert var.vtype in ["B", "C"], (
vtype = var.vtype
if vtype == "I":
assert var.ub == 1.0, (
"Only binary and continuous variables are currently supported. "
"Variable {var.varName} has type {var.vtype}."
"Integer variable {var.varName} has upper bound {var.ub}."
)
self._original_vtype[var] = var.vtype
if var.vtype == "B":
assert var.lb == 0.0, (
"Only binary and continuous variables are currently supported. "
"Integer variable {var.varName} has lower bound {var.ub}."
)
vtype = "B"
assert vtype in ["B", "C"], (
"Only binary and continuous variables are currently supported. "
"Variable {var.varName} has type {vtype}."
)
self._original_vtype[var] = vtype
if vtype == "B":
self._bin_vars.append(var)
def __getstate__(self) -> Dict: