swapped sdk to use pydantic dataclasses

Author: Gabefire

libs/labelbox/src/labelbox/orm/model.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from pydantic.dataclasses import dataclass
 from enum import Enum, auto
 from typing import TYPE_CHECKING, Any, Dict, List, Type, Union
 

libs/labelbox/src/labelbox/schema/export_task.py

@@ -2,7 +2,7 @@
 import os
 import tempfile
 from abc import ABC, abstractmethod
-from dataclasses import dataclass
+from pydantic.dataclasses import dataclass
 from enum import Enum
 from functools import lru_cache
 from typing import (

libs/labelbox/src/labelbox/schema/iam_integration.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from pydantic.dataclasses import dataclass
 
 from labelbox.utils import snake_case
 from labelbox.orm.db_object import DbObject

libs/labelbox/src/labelbox/schema/invite.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from pydantic.dataclasses import dataclass
 
 from labelbox.orm.db_object import DbObject
 from labelbox.orm.model import Field

libs/labelbox/src/labelbox/schema/ontology.py

@@ -1,17 +1,17 @@
 # type: ignore
-
+from dataclasses import field
 import colorsys
 import json
 import warnings
 from enum import Enum
 from typing import Annotated, Any, Dict, List, Optional, Type, Union
 
 from lbox.exceptions import InconsistentOntologyException
-from pydantic import StringConstraints, Field
+from pydantic import StringConstraints
 from pydantic.dataclasses import dataclass
 
 from labelbox.orm.db_object import DbObject
-from labelbox.orm.model import Field as LB_Field, Relationship
+from labelbox.orm.model import Field, Relationship
 from labelbox.schema.tool_building.step_reasoning_tool import StepReasoningTool
 from labelbox.schema.tool_building.tool_type import ToolType
 
@@ -35,9 +35,9 @@ def __str__(self):
 
 
 class FeatureSchema(DbObject):
-    name = LB_Field.String("name")
-    color = LB_Field.String("name")
-    normalized = LB_Field.Json("normalized")
+    name = Field.String("name")
+    color = Field.String("name")
+    normalized = Field.Json("normalized")
 
 
 @dataclass
@@ -64,7 +64,7 @@ class Option:
     feature_schema_id: Optional[FeatureSchemaId] = None
     options: Union[
         List["Classification"], List["PromptResponseClassification"]
-    ] = Field(default_factory=list)
+    ] = field(default_factory=list)
 
     def __post_init__(self):
         if self.label is None:
@@ -160,7 +160,7 @@ class UIMode(Enum):
     name: Optional[str] = None
     instructions: Optional[str] = None
     required: bool = False
-    options: List[Option] = Field(default_factory=list)
+    options: List[Option] = field(default_factory=list)
     schema_id: Optional[str] = None
     feature_schema_id: Optional[str] = None
     scope: Scope = None
@@ -344,7 +344,7 @@ class Type(Enum):
     name: Optional[str] = None
     instructions: Optional[str] = None
     required: bool = True
-    options: List[ResponseOption] = Field(default_factory=list)
+    options: List[ResponseOption] = field(default_factory=list)
     character_min: Optional[int] = None
     character_max: Optional[int] = None
     schema_id: Optional[str] = None
@@ -449,7 +449,7 @@ class Type(Enum):
     name: str
     required: bool = False
     color: Optional[str] = None
-    classifications: List[Classification] = Field(default_factory=list)
+    classifications: List[Classification] = field(default_factory=list)
     schema_id: Optional[str] = None
     feature_schema_id: Optional[str] = None
 
@@ -517,13 +517,13 @@ class Ontology(DbObject):
         created_by (Relationship): `ToOne` relationship to User
     """
 
-    name = LB_Field.String("name")
-    description = LB_Field.String("description")
-    updated_at = LB_Field.DateTime("updated_at")
-    created_at = LB_Field.DateTime("created_at")
-    normalized = LB_Field.Json("normalized")
-    object_schema_count = LB_Field.Int("object_schema_count")
-    classification_schema_count = LB_Field.Int("classification_schema_count")
+    name = Field.String("name")
+    description = Field.String("description")
+    updated_at = Field.DateTime("updated_at")
+    created_at = Field.DateTime("created_at")
+    normalized = Field.Json("normalized")
+    object_schema_count = Field.Int("object_schema_count")
+    classification_schema_count = Field.Int("classification_schema_count")
 
     projects = Relationship.ToMany("Project", True)
     created_by = Relationship.ToOne("User", False, "created_by")

libs/labelbox/src/labelbox/schema/role.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from pydantic.dataclasses import dataclass
 from typing import Dict, Optional, TYPE_CHECKING
 
 from labelbox.orm.model import Field

libs/labelbox/src/labelbox/schema/slice.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from pydantic.dataclasses import dataclass
 from typing import Optional, Tuple, Union
 import warnings
 from labelbox.orm.db_object import DbObject

libs/labelbox/src/labelbox/schema/tool_building/step_reasoning_tool.py

@@ -1,5 +1,6 @@
 import warnings
-from dataclasses import dataclass, field
+from dataclasses import field
+from pydantic.dataclasses import dataclass
 from typing import Any, Dict, List, Optional
 
 from labelbox.schema.tool_building.tool_type import ToolType