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

Fix mypy issues

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-30 11:55:08 -05:00
parent c3d26a1c75
commit 249002dcf3
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
miplearn/solvers/gurobi.py
View File

@@ -139,6 +139,7 @@ class GurobiSolver(InternalSolver):
) )
def enforce_constraints(self, names: List[str]) -> None: def enforce_constraints(self, names: List[str]) -> None:
assert self.model is not None
constr = [self._relaxed_constrs[n] for n in names] constr = [self._relaxed_constrs[n] for n in names]
for (i, (lhs, sense, rhs)) in enumerate(constr): for (i, (lhs, sense, rhs)) in enumerate(constr):
if sense == "=": if sense == "=":
@@ -425,7 +426,7 @@ class GurobiSolver(InternalSolver):
names: List[str], names: List[str],
tol: float = 1e-6, tol: float = 1e-6,
) -> List[bool]: ) -> List[bool]:
def _check(c): def _check(c: Tuple) -> bool:
lhs, sense, rhs = c lhs, sense, rhs = c
lhs_value = lhs.getValue() lhs_value = lhs.getValue()
if sense == "=": if sense == "=":
@@ -462,6 +463,7 @@ class GurobiSolver(InternalSolver):
return self.model.status in [self.gp.GRB.INFEASIBLE, self.gp.GRB.INF_OR_UNBD] return self.model.status in [self.gp.GRB.INFEASIBLE, self.gp.GRB.INF_OR_UNBD]
def relax_constraints(self, names: List[str]) -> None: def relax_constraints(self, names: List[str]) -> None:
assert self.model is not None
constrs = [self._cname_to_constr[n] for n in names] constrs = [self._cname_to_constr[n] for n in names]
for (i, name) in enumerate(names): for (i, name) in enumerate(names):
c = constrs[i] c = constrs[i]
@@ -707,7 +709,7 @@ class GurobiSolver(InternalSolver):
self._gp_vars = tuple(gp_vars) self._gp_vars = tuple(gp_vars)
self._gp_constrs = tuple(gp_constrs) self._gp_constrs = tuple(gp_constrs)
self._var_names = tuple(var_names) self._var_names = tuple(var_names)
self._constr_names = constr_names self._constr_names = tuple(constr_names)
self._var_types = tuple(var_types) self._var_types = tuple(var_types)
self._var_lbs = tuple(var_lbs) self._var_lbs = tuple(var_lbs)
self._var_ubs = tuple(var_ubs) self._var_ubs = tuple(var_ubs)