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

feat: JSON export can handle numpy.int64 entries

Browse source
This commit is contained in:
Simon Kuberski 2022-10-25 16:16:07 +02:00
parent e413bcefbc
commit 9ba46870d0
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pyerrors/input/json.py
View file

@ -186,10 +186,15 @@ def create_json_string(ol, description='', indent=1):
else:
raise Exception("Unkown datatype.")
def _jsonifier(o):
if isinstance(o, np.int64):
return int(o)
raise TypeError('%r is not JSON serializable' % o)
if indent:
return json.dumps(d, indent=indent, ensure_ascii=False, write_mode=json.WM_SINGLE_LINE_ARRAY)
return json.dumps(d, indent=indent, ensure_ascii=False, default=_jsonifier, write_mode=json.WM_SINGLE_LINE_ARRAY)
else:
return json.dumps(d, indent=indent, ensure_ascii=False, write_mode=json.WM_COMPACT)
return json.dumps(d, indent=indent, ensure_ascii=False, default=_jsonifier, write_mode=json.WM_COMPACT)
def dump_to_json(ol, fname, description='', indent=1, gz=True):