From 7ccb7875b9feb506e0ef83186bfe5fddc95db951 Mon Sep 17 00:00:00 2001
From: "Alinson S. Xavier" <git@axavier.org>
Date: Tue, 20 Aug 2024 16:46:20 -0500
Subject: [PATCH] Allow components to return stats, instead of modifying
 in-place

Added for compatibility with Julia.
---
 miplearn/solvers/learning.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/miplearn/solvers/learning.py b/miplearn/solvers/learning.py
index 21388e0..a4178e3 100644
--- a/miplearn/solvers/learning.py
+++ b/miplearn/solvers/learning.py
@@ -47,7 +47,9 @@ class LearningSolver:
                 relaxed.optimize()
                 relaxed.extract_after_lp(h5)
             for comp in self.components:
-                comp.before_mip(h5_filename, model, stats)
+                comp_stats = comp.before_mip(h5_filename, model, stats)
+                if comp_stats is not None:
+                    stats.update(comp_stats)
             model.optimize()
             model.extract_after_mip(h5)