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

GurobiPyomoSolver: Extract same features as GurobiSolver

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-11 17:05:41 -05:00
parent 6bc81417ac
commit 16630b3a36
3 changed files with 96 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -505,7 +505,10 @@ class BasePyomoSolver(InternalSolver):
self._varname_to_var = {}
for var in self.model.component_objects(Var):
for idx in var:
self._varname_to_var[f"{var.name}[{idx}]"] = var[idx]
varname = f"{var.name}[{idx}]"
if idx is None:
varname = var.name
self._varname_to_var[varname] = var[idx]
self._all_vars += [var[idx]]
if var[idx].domain == pyomo.core.base.set_types.Binary:
self._bin_vars += [var[idx]]