Extend NumberEncoder to encode also numpy.ndarray

miccoli · Aug 6, 2022 · 380ee7c · 380ee7c
1 parent f9fd8c7
commit 380ee7c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/zarr/util.py b/zarr/util.py
@@ -42,6 +42,12 @@ def default(self, o):
             return int(o)
         if isinstance(o, numbers.Real):
             return float(o)
+        # hdf5 attributes can be arrays, which h5py renders as np.ndarray
+        if isinstance(o, np.ndarray):
+            return o.tolist()
+        # this is for encoding numpy.bytes_
+        if isinstance(o, bytes):
+            return o.decode()
         return json.JSONEncoder.default(self, o)