Fix mypy errors

2025-12-06 01:18:52 -06:00 · 2021-04-13 14:36:20 -05:00
parent 61645491a4
commit 37a1bc9fe6
1 changed files with 7 additions and 0 deletions
--- a/miplearn/solvers/gurobi.py
+++ b/miplearn/solvers/gurobi.py
@@ -251,6 +251,13 @@ class GurobiSolver(InternalSolver):
            if values is not None:
                var.value = values[i]
            if rc is not None:
+                assert sa_obj_up is not None
+                assert sa_obj_down is not None
+                assert sa_ub_up is not None
+                assert sa_ub_down is not None
+                assert sa_lb_up is not None
+                assert sa_lb_down is not None
+                assert vbasis is not None
                var.reduced_cost = rc[i]
                var.sa_obj_up = sa_obj_up[i]
                var.sa_obj_down = sa_obj_down[i]