diff options
Diffstat (limited to 'scripts/json2numpy/json2numpy.py')
-rw-r--r-- | scripts/json2numpy/json2numpy.py | 4 |
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) |