Add training_data argument to after_solve

2025-12-08 02:18:51 -06:00 · 2021-01-14 10:37:48 -06:00
parent 30d6ea0a9b
commit e12a896504
15 changed files with 148 additions and 58 deletions
--- a/miplearn/components/primal.py
+++ b/miplearn/components/primal.py
@@ -39,7 +39,14 @@ class PrimalSolutionComponent(Component):
        else:
            solver.internal_solver.set_warm_start(solution)

-    def after_solve(self, solver, instance, model, results):
+    def after_solve(
+        self,
+        solver,
+        instance,
+        model,
+        stats,
+        training_data,
+    ):
        pass

    def x(self, training_instances):