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

Store cuts and lazy constraints as JSON in H5

Browse Source
This commit is contained in:
Alinson S. Xavier
2024-02-01 10:06:21 -06:00
parent 2774edae8c
commit 281508f44c
41 changed files with 30 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
miplearn/solvers/gurobi.py
View File

@@ -1,7 +1,9 @@
# MIPLearn: Extensible Framework for Learning-Enhanced Mixed-Integer Optimization
# Copyright (C) 2020-2022, UChicago Argonne, LLC. All rights reserved.
# Released under the modified BSD license. See COPYING.md for more details.
import logging
import json
from typing import Dict, Optional, Callable, Any, List
import gurobipy as gp
@@ -167,9 +169,9 @@ class GurobiModel(AbstractModel):
pass
self._extract_after_mip_solution_pool(h5)
if self.lazy_ is not None:
h5.put_scalar("mip_lazy", repr(self.lazy_))
h5.put_scalar("mip_lazy", json.dumps(self.lazy_))
if self.cuts_ is not None:
h5.put_scalar("mip_cuts", repr(self.cuts_))
h5.put_scalar("mip_cuts", json.dumps(self.cuts_))
def fix_variables(
self,