changed to pydantic dataclasses

Author: Gabefire

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

@@ -3,15 +3,15 @@
 import colorsys
 import json
 import warnings
-from dataclasses import dataclass, field
 from enum import Enum
 from typing import Annotated, Any, Dict, List, Optional, Type, Union
 
 from lbox.exceptions import InconsistentOntologyException
-from pydantic import StringConstraints
+from pydantic import StringConstraints, Field
+from pydantic.dataclasses import dataclass
 
 from labelbox.orm.db_object import DbObject
-from labelbox.orm.model import Field, Relationship
+from labelbox.orm.model import Field as LB_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 = Field.String("name")
-    color = Field.String("name")
-    normalized = Field.Json("normalized")
+    name = LB_Field.String("name")
+    color = LB_Field.String("name")
+    normalized = LB_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 = 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")
+    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")
 
     projects = Relationship.ToMany("Project", True)
     created_by = Relationship.ToOne("User", False, "created_by")
@@ -596,10 +596,10 @@ class OntologyBuilder:
 
     """
 
-    tools: List[Union[Tool, StepReasoningTool]] = field(default_factory=list)
+    tools: List[Union[Tool, StepReasoningTool]] = Field(default_factory=list)
     classifications: List[
         Union[Classification, PromptResponseClassification]
-    ] = field(default_factory=list)
+    ] = Field(default_factory=list)
 
     @classmethod
     def from_dict(cls, dictionary: Dict[str, Any]) -> Dict[str, Any]: