FileInstance.save: create file when it does not already exist

This commit is contained in:
2021-07-27 11:22:40 -05:00
parent 4f14b99a75
commit d30c3232e6
3 changed files with 8 additions and 8 deletions

View File

@@ -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]:

View File

@@ -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)