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

Require a callable as the internal solver

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-01-19 21:21:39 -06:00
parent 3ff773402d
commit ffc77075f5
5 changed files with 57 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
miplearn/components/steps/tests/test_convert_tight.py
View File

@@ -13,7 +13,7 @@ def test_convert_tight_usage():
capacity=16.0,
)
solver = LearningSolver(
solver=GurobiSolver(),
solver=GurobiSolver,
components=[
RelaxIntegralityStep(),
ConvertTightIneqsIntoEqsStep(),
@@ -64,7 +64,7 @@ def test_convert_tight_infeasibility():
comp.classifiers["c3"].predict_proba = Mock(return_value=[[1, 0]])
solver = LearningSolver(
solver=GurobiSolver(params={}),
solver=GurobiSolver,
components=[comp],
solve_lp_first=False,
)
@@ -87,7 +87,7 @@ def test_convert_tight_suboptimality():
comp.classifiers["c3"].predict_proba = Mock(return_value=[[0, 1]])
solver = LearningSolver(
solver=GurobiSolver(params={}),
solver=GurobiSolver,
components=[comp],
solve_lp_first=False,
)
@@ -110,7 +110,7 @@ def test_convert_tight_optimal():
comp.classifiers["c3"].predict_proba = Mock(return_value=[[0, 1]])
solver = LearningSolver(
solver=GurobiSolver(params={}),
solver=GurobiSolver,
components=[comp],
solve_lp_first=False,
)