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

Fix failing tests

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-04-07 09:41:30 -05:00
parent b35f411199
commit af4a6d6a4e
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/python/miplearn/benchmark.py
View File

@@ -28,8 +28,7 @@ class BenchmarkRunner:
for (name, solver) in self.solvers.items(): for (name, solver) in self.solvers.items():
results = solver.parallel_solve(instances, results = solver.parallel_solve(instances,
n_jobs=n_jobs, n_jobs=n_jobs,
label="Solve (%s)" % name, label="Solve (%s)" % name)
collect_training_data=False)
for i in range(len(instances)): for i in range(len(instances)):
self._push_result(results[i], self._push_result(results[i],
solver=solver, solver=solver,

3
src/python/miplearn/tests/test_solver.py
View File

@@ -2,6 +2,9 @@
# Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved. # Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
# Released under the modified BSD license. See COPYING.md for more details. # Released under the modified BSD license. See COPYING.md for more details.
import pickle
import tempfile
from miplearn import LearningSolver, BranchPriorityComponent from miplearn import LearningSolver, BranchPriorityComponent
from miplearn.problems.knapsack import KnapsackInstance from miplearn.problems.knapsack import KnapsackInstance
from miplearn.solvers import GurobiSolver from miplearn.solvers import GurobiSolver