aboutsummaryrefslogtreecommitdiff
path: root/scripts/json2numpy/json2numpy.py
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/json2numpy/json2numpy.py')
-rw-r--r--scripts/json2numpy/json2numpy.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/scripts/json2numpy/json2numpy.py b/scripts/json2numpy/json2numpy.py
index 21b1acd..c04013e 100644
--- a/scripts/json2numpy/json2numpy.py
+++ b/scripts/json2numpy/json2numpy.py
@@ -16,8 +16,10 @@ class NumpyArrayEncoder(json.JSONEncoder):
"""Encode default operation."""
if isinstance(obj, np.integer):
return int(obj)
- elif isinstance(obj, np.floating):
+ elif isinstance(obj, np.float32):
return float(obj)
+ elif isinstance(obj, np.float16):
+ return np.float16(obj)
elif isinstance(obj, np.ndarray):
return obj.tolist()
return super(NumpyArrayEncoder, self).default(obj)