From 2bd2c7fbc847ec296d88f5409943895c4b127ef4 Mon Sep 17 00:00:00 2001 From: Gabefire <33893811+Gabefire@users.noreply.github.com> Date: Mon, 9 Sep 2024 12:09:34 -0500 Subject: [PATCH] removed protected messages to avoid unneccassary warnings to users --- .../src/labelbox/data/annotation_types/data/raster.py | 3 ++- libs/labelbox/src/labelbox/data/annotation_types/mmc.py | 6 +++++- libs/labelbox/src/labelbox/schema/foundry/app.py | 2 ++ 3 files changed, 9 insertions(+), 2 deletions(-) diff --git a/libs/labelbox/src/labelbox/data/annotation_types/data/raster.py b/libs/labelbox/src/labelbox/data/annotation_types/data/raster.py index 50c7f4947..234d8b136 100644 --- a/libs/labelbox/src/labelbox/data/annotation_types/data/raster.py +++ b/libs/labelbox/src/labelbox/data/annotation_types/data/raster.py @@ -24,7 +24,8 @@ class RasterData(BaseModel, ABC): uid: Optional[str] = None global_key: Optional[str] = None arr: Optional[TypedArray[Literal['uint8']]] = None - model_config = ConfigDict(extra="forbid", copy_on_model_validation="none") + + model_config = ConfigDict(extra="forbid") @classmethod def from_2D_arr(cls, arr: Union[TypedArray[Literal['uint8']], diff --git a/libs/labelbox/src/labelbox/data/annotation_types/mmc.py b/libs/labelbox/src/labelbox/data/annotation_types/mmc.py index 8c4ca0d3e..d3ab763cb 100644 --- a/libs/labelbox/src/labelbox/data/annotation_types/mmc.py +++ b/libs/labelbox/src/labelbox/data/annotation_types/mmc.py @@ -1,7 +1,7 @@ from abc import ABC from typing import ClassVar, List, Union -from pydantic import field_validator +from pydantic import ConfigDict, field_validator from labelbox.utils import _CamelCaseMixin from labelbox.data.annotation_types.annotation import BaseAnnotation @@ -10,6 +10,8 @@ class MessageInfo(_CamelCaseMixin): message_id: str model_config_name: str + + model_config = ConfigDict(protected_namespaces=()) class OrderedMessageInfo(MessageInfo): @@ -19,6 +21,8 @@ class OrderedMessageInfo(MessageInfo): class _BaseMessageEvaluationTask(_CamelCaseMixin, ABC): format: ClassVar[str] parent_message_id: str + + model_config = ConfigDict(protected_namespaces=()) class MessageSingleSelectionTask(_BaseMessageEvaluationTask, MessageInfo): diff --git a/libs/labelbox/src/labelbox/schema/foundry/app.py b/libs/labelbox/src/labelbox/schema/foundry/app.py index 52743e55b..f73d5056f 100644 --- a/libs/labelbox/src/labelbox/schema/foundry/app.py +++ b/libs/labelbox/src/labelbox/schema/foundry/app.py @@ -13,6 +13,8 @@ class App(_CamelCaseMixin): class_to_schema_id: Dict[str, str] ontology_id: str created_by: Optional[str] = None + + model_config = ConfigDict(protected_namespaces=()) @classmethod def type_name(cls):