lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-11-04 01:25:46 +01:00
Code Issues Projects Releases Packages Wiki Activity

fix: Obs.dump can now be provided with a description for the json.gz

Browse source 
file
This commit is contained in:
Fabian Joswig 2022-02-13 16:21:27 +00:00
commit e80fde6630
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
pyerrors/obs.py
View file

@ -597,7 +597,7 @@ class Obs:
return dict(zip(self.e_names, sizes))
def dump(self, filename, datatype="json.gz", **kwargs):
def dump(self, filename, datatype="json.gz", description="", **kwargs):
"""Dump the Obs to a file 'name' of chosen format.
Parameters
@ -607,6 +607,8 @@ class Obs:
datatype : str
Format of the exported file. Supported formats include
"json.gz" and "pickle"
description : str
Description for output file, only relevant for json.gz format.
path : str
specifies a custom path for the file (default '.')
"""
@ -617,7 +619,7 @@ class Obs:
if datatype == "json.gz":
from .input.json import dump_to_json
dump_to_json([self], file_name)
dump_to_json([self], file_name, description=description)
elif datatype == "pickle":
with open(file_name + '.p', 'wb') as fb:
pickle.dump(self, fb)