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

simulate_perfect: Do not overwrite original file

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-01-13 11:04:33 -06:00
parent b01d97cc2b
commit beee252fa2
6 changed files with 46 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
miplearn/extractors.py
View File

@@ -29,12 +29,15 @@ class InstanceIterator:
self.current += 1
if isinstance(result, str):
logger.debug("Read: %s" % result)
if result.endswith(".gz"):
with gzip.GzipFile(result, "rb") as file:
result = pickle.load(file)
else:
with open(result, "rb") as file:
result = pickle.load(file)
try:
if result.endswith(".gz"):
with gzip.GzipFile(result, "rb") as file:
result = pickle.load(file)
else:
with open(result, "rb") as file:
result = pickle.load(file)
except pickle.UnpicklingError:
raise Exception(f"Invalid instance file: {result}")
return result