mirror of
https://github.com/ANL-CEEESA/MIPLearn.git
synced 2025-12-06 09:28:51 -06:00
FileInstance.save: create file when it does not already exist
This commit is contained in:
@@ -167,8 +167,8 @@ class Hdf5Sample(Sample):
|
||||
are actually accessed, and therefore it is more scalable.
|
||||
"""
|
||||
|
||||
def __init__(self, filename: str) -> None:
|
||||
self.file = h5py.File(filename, "r+")
|
||||
def __init__(self, filename: str, mode: str = "r+") -> None:
|
||||
self.file = h5py.File(filename, mode)
|
||||
|
||||
@overrides
|
||||
def get_bytes(self, key: str) -> Optional[bytes]:
|
||||
|
||||
@@ -119,7 +119,7 @@ class FileInstance(Instance):
|
||||
|
||||
@classmethod
|
||||
def save(cls, instance: Instance, filename: str) -> None:
|
||||
h5 = Hdf5Sample(filename)
|
||||
h5 = Hdf5Sample(filename, mode="w")
|
||||
instance_pkl = pickle.dumps(instance)
|
||||
h5.put_bytes("pickled", instance_pkl)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user