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

Fix mypy errors

Browse Source
This commit is contained in:
Alinson S. Xavier
2023-10-26 13:39:57 -05:00
parent e555dffc0c
commit 2d07a44f7d
12 changed files with 61 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/problems/test_setcover.py
View File

@@ -14,6 +14,7 @@ from miplearn.problems.setcover import (
SetCoverGenerator,
build_setcover_model_pyomo,
)
from miplearn.solvers.abstract import AbstractModel
def test_set_cover_generator() -> None:
@@ -84,6 +85,7 @@ def test_set_cover() -> None:
build_setcover_model_pyomo(data),
build_setcover_model_gurobipy(data),
]:
assert isinstance(model, AbstractModel)
with NamedTemporaryFile() as tempfile:
with H5File(tempfile.name) as h5:
model.optimize()

2
tests/problems/test_stab.py
View File

@@ -12,6 +12,7 @@ from miplearn.problems.stab import (
build_stab_model_pyomo,
build_stab_model_gurobipy,
)
from miplearn.solvers.abstract import AbstractModel
def test_stab() -> None:
@@ -23,6 +24,7 @@ def test_stab() -> None:
build_stab_model_pyomo(data),
build_stab_model_gurobipy(data),
]:
assert isinstance(model, AbstractModel)
with NamedTemporaryFile() as tempfile:
with H5File(tempfile.name) as h5:
model.optimize()