Remove unused classes and methods

This commit is contained in:
2021-05-15 14:29:11 -05:00
parent 91c8db2225
commit 4bf4d09cb5
11 changed files with 19 additions and 195 deletions

View File

@@ -10,7 +10,7 @@ from pyomo import environ as pe
from scipy.stats import randint
from miplearn.solvers.pyomo.base import BasePyomoSolver
from miplearn.types import SolverParams, BranchPriorities
from miplearn.types import SolverParams
logger = logging.getLogger(__name__)
@@ -42,17 +42,6 @@ class GurobiPyomoSolver(BasePyomoSolver):
def clone(self) -> "GurobiPyomoSolver":
return GurobiPyomoSolver(params=self.params)
@overrides
def set_branching_priorities(self, priorities: BranchPriorities) -> None:
from gurobipy import GRB
for (varname, priority) in priorities.items():
if priority is None:
continue
var = self._varname_to_var[varname]
gvar = self._pyomo_solver._pyomo_var_to_solver_var_map[var]
gvar.setAttr(GRB.Attr.BranchPriority, int(round(priority)))
@overrides
def _extract_node_count(self, log: str) -> int:
return max(1, int(self._pyomo_solver._solver_model.getAttr("NodeCount")))