fix: support numpy scalars

widgetti · Oct 17, 2023 · 63e05bb · 63e05bb
1 parent a375a5a
commit 63e05bb
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/solara/server/kernel.py b/solara/server/kernel.py
@@ -49,6 +49,13 @@ def json_default(obj):
         return obj.isoformat().replace("+00:00", "Z")
     elif isinstance(obj, bytes):
         return b2a_base64(obj).decode("ascii")
+    if type(obj).__module__ == "numpy":
+        import numpy as np
+
+        if isinstance(obj, np.number):
+            return repr(obj.item())
+        else:
+            raise TypeError("%r is not JSON serializable" % obj)
     else:
         raise TypeError("%r is not JSON serializable" % obj)