From 6256abdd9744d77d81144b357830925d9b0a7b0c Mon Sep 17 00:00:00 2001 From: Val Brodsky Date: Tue, 15 Oct 2024 10:01:30 -0700 Subject: [PATCH] Fix labeling dashboard serialization --- .../src/labelbox/schema/labeling_service_dashboard.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/libs/labelbox/src/labelbox/schema/labeling_service_dashboard.py b/libs/labelbox/src/labelbox/schema/labeling_service_dashboard.py index 2f91af7af..b8e2fe226 100644 --- a/libs/labelbox/src/labelbox/schema/labeling_service_dashboard.py +++ b/libs/labelbox/src/labelbox/schema/labeling_service_dashboard.py @@ -3,7 +3,7 @@ from typing import Any, Dict, List, Optional, Union from lbox.exceptions import ResourceNotFoundError -from pydantic import BaseModel, Field, model_validator, model_serializer +from pydantic import BaseModel, Field, model_serializer, model_validator from labelbox.pagination import PaginatedCollection from labelbox.schema.labeling_service_status import LabelingServiceStatus @@ -221,9 +221,9 @@ def convert_boost_data(cls, data): return data - @model_serializer() - def ser_model(self): - row = self + @model_serializer(mode="wrap") + def ser_model(self, handler): + row = handler(self) row.pop("client") row["service_type"] = self.service_type return row