ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-07 18:08:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

Move user_cuts/lazy_enforced to sample.data

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-07-01 08:37:31 -05:00
parent 80281df8d8
commit 061b1349fe
10 changed files with 56 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
miplearn/components/dynamic_user_cuts.py
View File

@@ -87,9 +87,7 @@ class UserCutsComponent(Component):
stats: LearningSolveStats,
sample: Sample,
) -> None:
assert sample.after_mip is not None
assert sample.after_mip.extra is not None
sample.after_mip.extra["user_cuts_enforced"] = set(self.enforced)
sample.put("user_cuts_enforced", set(self.enforced))
stats["UserCuts: Added in callback"] = self.n_added_in_callback
if self.n_added_in_callback > 0:
logger.info(f"{self.n_added_in_callback} user cuts added in callback")