fixed Pydantic warning

aponcedeleonch · aponcedeleonch · commit 714a63066da8 · 2025-03-07T11:10:20.000+02:00
diff --git a/src/codegate/db/models.py b/src/codegate/db/models.py
@@ -276,7 +276,7 @@ def nd_array_custom_serializer(x):
 NdArray = Annotated[
     np.ndarray,
     BeforeValidator(nd_array_custom_before_validator),
-    PlainSerializer(nd_array_custom_serializer, return_type=str),
+    PlainSerializer(nd_array_custom_serializer, return_type=np.ndarray),
 ]
 
 VALID_PERSONA_NAME_PATTERN = re.compile(r"^[a-zA-Z0-9_ -]+$")

Original file line number	Diff line number	Diff line change
`@@ -276,7 +276,7 @@ def nd_array_custom_serializer(x):`
`276`	`276`	`NdArray = Annotated[`
`277`	`277`	`np.ndarray,`
`278`	`278`	`BeforeValidator(nd_array_custom_before_validator),`
`279`		`- PlainSerializer(nd_array_custom_serializer, return_type=str),`
	`279`	`+ PlainSerializer(nd_array_custom_serializer, return_type=np.ndarray),`
`280`	`280`	`]`
`281`	`281`
`282`	`282`	`VALID_PERSONA_NAME_PATTERN = re.compile(r"^[a-zA-Z0-9_ -]+$")`