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

Add types to solvers

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-07 20:58:44 -05:00
parent 38212fb858
commit 331ee5914d
5 changed files with 35 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -228,7 +228,7 @@ class BasePyomoSolver(InternalSolver):
self._pyomo_solver.update_var(var)
@overrides
def add_constraint(self, constraint):
def add_constraint(self, constraint: Any) -> Any:
self._pyomo_solver.add_constraint(constraint)
self._update_constrs()
@@ -261,7 +261,7 @@ class BasePyomoSolver(InternalSolver):
return int(value)
@overrides
def get_constraint_ids(self):
def get_constraint_ids(self) -> List[str]:
return list(self._cname_to_constr.keys())
def _get_warm_start_regexp(self) -> Optional[str]:
@@ -332,7 +332,7 @@ class BasePyomoSolver(InternalSolver):
return self._termination_condition == TerminationCondition.infeasible
@overrides
def get_dual(self, cid):
def get_dual(self, cid: str) -> float:
raise NotImplementedError()
@overrides