From 71206a03fdcb147eab21b0b4c8ef5cba09967cd1 Mon Sep 17 00:00:00 2001 From: Christoph Eder Date: Fri, 12 Dec 2025 16:59:57 +0100 Subject: [PATCH] Changes in embedding.py Another try, this time trusting pydantic_tensor to fix it --- anansyTypes/structures/embedding.py | 25 +------------------------ 1 file changed, 1 insertion(+), 24 deletions(-) diff --git a/anansyTypes/structures/embedding.py b/anansyTypes/structures/embedding.py index d6851b8..a080971 100644 --- a/anansyTypes/structures/embedding.py +++ b/anansyTypes/structures/embedding.py @@ -22,27 +22,4 @@ class Embedding: # Fallback or error if not implemented for other types (e.g., torch.Tensor) raise NotImplementedError( "Comparison must be implemented for all supported data types." - ) - - @classmethod - def __get_pydantic_core_schema__(cls, source_type: Any, handler): - def serialize(value: "Embedding") -> dict: - serialized_data = handler(source_type=value.data.__class__, value=value.data) - return { - 'data': serialized_data - } - - return core_schema.json_or_python_schema( - python_schema=core_schema.dataclass_schema( - schema=handler.generate_schema(cls), - fields={ - 'data': core_schema.model_field( - schema=handler.generate_schema(cls.data), - required=True - ), - } - ), - json_schema=core_schema.plain_serializer_function_ser_schema( - serialize, when_used='json' - ), - ) + ) \ No newline at end of file