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

Remove most usages of put_{vector,vector_list}; deprecate get_set

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-10 11:52:02 -05:00
parent 60b9a6775f
commit ed58242b5c
9 changed files with 40 additions and 176 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
miplearn/instance/file.py
View File

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