changed encoding of symbols

This commit is contained in:
stuart committed 2025-12-16 11:41:31 +01:00
1 parent ed16b208d7
commit 6dd0a4d313
1 file changed
+10 -12
+10 -12
View File
@@ -15,18 +15,18 @@ class Symbol:
@classmethod
def from_dict(cls, data: dict) -> "Symbol":
if isinstance(data["image"], str):
# Decode from base64
image_bytes = base64.b64decode(data["image"])
image_array = np.frombuffer(image_bytes, dtype=np.uint8)
else:
# From list/array
image_array = np.array(data["image"], dtype=np.int32)
# if isinstance(data["image"], str):
# # Decode from base64
# image_bytes = base64.b64decode(data["image"])
# image_array = np.frombuffer(image_bytes, dtype=np.uint8)
# else:
# # From list/array
# image_array = np.array(data["image"], dtype=np.int32)
image_data=data["image"].encode("utf-8")
return cls(
id=data["id"],
label=data["label"],
image=image_array,
image=image_data,
)
@classmethod
@@ -46,9 +46,7 @@ class Symbol:
lambda x: {
"id": x.id,
"label": x.label,
"image": base64.b64encode(np.array(x.image).tobytes()).decode(
"utf-8"
),
"image": x.image.decode("utf-8"),
},
when_used="json",
),