support ndarray in serializer

gizatechxyz · May 3, 2024 · cf987e9 · cf987e9
1 parent 06dc8a0
commit cf987e9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/osiris/cairo/serde/serialize.py b/osiris/cairo/serde/serialize.py
@@ -4,7 +4,7 @@
     Tensor,
 )
 from osiris.cairo.serde.utils import int_to_felt
-
+import numpy as np
 
 def serializer(data):
     if isinstance(data, bool):
@@ -16,10 +16,11 @@ def serializer(data):
     elif isinstance(data, (list, tuple)):
         joined_elements = ' '.join(serializer(e) for e in data)
         return f"[{joined_elements}]"
+    elif isinstance(data, np.ndarray):
+        return f"{serializer(data.tolist())}"
     elif isinstance(data, Tensor):
         return f"{serializer(data.shape)} {serializer(data.data)}"
     elif isinstance(data, FixedPoint):
         return f"{serializer(data.mag)} {serializer(data.sign)}"
-
     else:
         raise ValueError("Unsupported data type for serialization")