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

Re-add sample.{get,put}_bytes

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-11 06:24:10 -05:00
parent 256d3d094f
commit 5b3a56f053
4 changed files with 25 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
miplearn/instance/file.py
View File

@@ -111,14 +111,16 @@ class FileInstance(Instance):
def load(self) -> None:
if self.instance is not None:
return
self.instance = pickle.loads(self.h5.get_array("pickled").tobytes())
pkl = self.h5.get_bytes("pickled")
assert pkl is not None
self.instance = pickle.loads(pkl)
assert isinstance(self.instance, Instance)
@classmethod
def save(cls, instance: Instance, filename: str) -> None:
h5 = Hdf5Sample(filename, mode="w")
instance_pkl = np.frombuffer(pickle.dumps(instance), dtype=np.int8)
h5.put_array("pickled", instance_pkl)
instance_pkl = pickle.dumps(instance)
h5.put_bytes("pickled", instance_pkl)
@overrides
def create_sample(self) -> Sample: