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

Always remove .mypy_cache; fix more mypy tests

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-09 08:18:54 -05:00
parent 32b6a8f3fa
commit 3f4336f902
3 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_benchmark.py
View File

@@ -20,7 +20,7 @@ def test_benchmark() -> None:
# Solve training instances
training_solver = LearningSolver()
training_solver.parallel_solve(train_instances, n_jobs=n_jobs)
training_solver.parallel_solve(train_instances, n_jobs=n_jobs) # type: ignore
# Benchmark
test_solvers = {
@@ -28,8 +28,12 @@ def test_benchmark() -> None:
"Strategy B": LearningSolver(),
}
benchmark = BenchmarkRunner(test_solvers)
benchmark.fit(train_instances)
benchmark.parallel_solve(test_instances, n_jobs=n_jobs, n_trials=2)
benchmark.fit(train_instances) # type: ignore
benchmark.parallel_solve(
test_instances, # type: ignore
n_jobs=n_jobs,
n_trials=2,
)
assert benchmark.results.values.shape == (12, 20)
benchmark.write_csv("/tmp/benchmark.csv")