Fix mypy errors

This commit is contained in:
2023-10-26 13:39:57 -05:00
parent e555dffc0c
commit 2d07a44f7d
12 changed files with 61 additions and 41 deletions

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()