[PLT-1725] Changed ontology dataclasses to pydantic dataclasses

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/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

@@ -2,7 +2,8 @@
 
 import colorsys
 import json
-from dataclasses import dataclass, field
+from pydantic.dataclasses import dataclass
+from dataclasses import field
 from enum import Enum
 from typing import Any, Dict, List, Optional, Union
 

libs/labelbox/src/labelbox/schema/project.py

@@ -1412,7 +1412,7 @@ def move_data_rows_to_task_queue(
         Args:
             data_row_ids: a list of data row ids to be moved. This should be a DataRowIdentifiers object
                 DataRowIdentifier objects are lists of ids or global keys. A DataIdentifier object can be a UniqueIds or GlobalKeys class.
-            task_queue_id: the task queue id to be moved to, or None to specify the "Done" queue
+            task_queue_id: the task queue id to be moved to, or None to specify the "Done" queue.
 
         Returns:
             None if successful, or a raised error on failure

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/tool_building/base_step_reasoning_tool.py

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

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

@@ -1,5 +1,6 @@
 import warnings
-from dataclasses import dataclass, field
+from pydantic.dataclasses import dataclass
+from dataclasses import field
 from enum import Enum
 from typing import Any, Dict, List, Optional, Union
 

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

@@ -1,4 +1,5 @@
-from dataclasses import dataclass, field
+from pydantic.dataclasses import dataclass
+from dataclasses import field
 from enum import Enum
 
 from labelbox.schema.tool_building.base_step_reasoning_tool import (

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

@@ -1,4 +1,5 @@
-from dataclasses import dataclass, field
+from pydantic.dataclasses import dataclass
+from dataclasses import field
 from typing import Any, Dict, List, Optional
 
 from labelbox.schema.tool_building.classification import (

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

@@ -1,4 +1,5 @@
-from dataclasses import dataclass, field
+from pydantic.dataclasses import dataclass
+from dataclasses import field
 from enum import Enum
 
 from labelbox.schema.tool_building.base_step_reasoning_tool import (

libs/labelbox/src/labelbox/schema/user_group_v2.py

@@ -1,6 +1,6 @@
 import json
 import warnings
-from dataclasses import dataclass
+from pydantic.dataclasses import dataclass
 from io import BytesIO
 from typing import List, Optional