Fix mypy errors

2025-12-06 01:18:52 -06:00 · 2021-04-13 06:47:31 -05:00
parent 9d404f29a7
commit d7aa31f3eb
2 changed files with 11 additions and 1 deletions
--- a/tests/components/test_static_lazy.py
+++ b/tests/components/test_static_lazy.py
@@ -114,6 +114,10 @@ def test_usage_with_solver(instance: Instance) -> None:

    stats: LearningSolveStats = {}
    sample = instance.samples[0]
+    assert sample.after_load is not None
+    assert sample.after_load.constraints is not None
+    assert sample.after_mip is not None
+    assert sample.after_mip.extra is not None
    del sample.after_mip.extra["lazy_enforced"]

    # LearningSolver calls before_solve_mip