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

Add first model feature (constraint RHS)

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-03-02 17:21:05 -06:00
parent 31ca45036a
commit 1397937f03
17 changed files with 167 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
miplearn/solvers/pyomo/base.py
View File

@@ -212,7 +212,11 @@ class BasePyomoSolver(InternalSolver):
assert self.model is not None
self._cname_to_constr = {}
for constr in self.model.component_objects(Constraint):
self._cname_to_constr[constr.name] = constr
if isinstance(constr, pe.ConstraintList):
for idx in constr:
self._cname_to_constr[f"{constr.name}[{idx}]"] = constr[idx]
else:
self._cname_to_constr[constr.name] = constr
def fix(self, solution):
count_total, count_fixed = 0, 0
@@ -302,6 +306,13 @@ class BasePyomoSolver(InternalSolver):
else:
return "="
def get_constraint_rhs(self, cid: str) -> float:
cobj = self._cname_to_constr[cid]
if cobj.has_ub:
return cobj.upper()
else:
return cobj.lower()
def set_constraint_sense(self, cid: str, sense: str) -> None:
raise Exception("Not implemented")