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

BasicCollector: Do not crash on exception

Browse Source
This commit is contained in:
Alinson S. Xavier
2024-02-26 16:41:50 -06:00
parent 8a02e22a35
commit 0534d50af3
2 changed files with 49 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
miplearn/collectors/basic.py
View File

@@ -9,6 +9,7 @@ import sys
from io import StringIO
from os.path import exists
from typing import Callable, List, Any
import traceback
from ..h5 import H5File
from ..io import _RedirectOutput, gzip, _to_h5_filename
@@ -29,6 +30,7 @@ class BasicCollector:
verbose: bool = False,
) -> None:
def _collect(data_filename: str) -> None:
try:
h5_filename = _to_h5_filename(data_filename)
mps_filename = h5_filename.replace(".h5", ".mps")
@@ -75,6 +77,10 @@ class BasicCollector:
gzip(mps_filename)
h5.put_scalar("mip_log", streams[0].getvalue())
except:
print(f"Error processing: data_filename")
traceback.print_exc()
if n_jobs > 1:
p_umap(

7
miplearn/io.py
View File

@@ -87,7 +87,10 @@ def read_pkl_gz(filename: str) -> Any:
def _to_h5_filename(data_filename: str) -> str:
output = f"{data_filename}.h5"
output = output.replace(".gz.h5", ".h5")
output = output.replace(".json.h5", ".h5")
output = output.replace(".pkl.h5", ".h5")
output = output.replace(".csv.h5", ".h5")
output = output.replace(".jld2.h5", ".h5")
output = output.replace(".json.h5", ".h5")
output = output.replace(".lp.h5", ".h5")
output = output.replace(".mps.h5", ".h5")
output = output.replace(".pkl.h5", ".h5")
return output