diff --git a/libs/labelbox/src/labelbox/orm/model.py b/libs/labelbox/src/labelbox/orm/model.py
index 535ab0f7d..10c594d89 100644
--- a/libs/labelbox/src/labelbox/orm/model.py
+++ b/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
 
diff --git a/libs/labelbox/src/labelbox/schema/iam_integration.py b/libs/labelbox/src/labelbox/schema/iam_integration.py
index cb5309929..7c680b781 100644
--- a/libs/labelbox/src/labelbox/schema/iam_integration.py
+++ b/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
diff --git a/libs/labelbox/src/labelbox/schema/invite.py b/libs/labelbox/src/labelbox/schema/invite.py
index c89a8b08c..7b73d943a 100644
--- a/libs/labelbox/src/labelbox/schema/invite.py
+++ b/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
diff --git a/libs/labelbox/src/labelbox/schema/ontology.py b/libs/labelbox/src/labelbox/schema/ontology.py
index 0032aaad1..189b4c82d 100644
--- a/libs/labelbox/src/labelbox/schema/ontology.py
+++ b/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
 
diff --git a/libs/labelbox/src/labelbox/schema/project.py b/libs/labelbox/src/labelbox/schema/project.py
index 929afc729..8f1d76943 100644
--- a/libs/labelbox/src/labelbox/schema/project.py
+++ b/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
diff --git a/libs/labelbox/src/labelbox/schema/role.py b/libs/labelbox/src/labelbox/schema/role.py
index d22e2a78e..5393a88cc 100644
--- a/libs/labelbox/src/labelbox/schema/role.py
+++ b/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
diff --git a/libs/labelbox/src/labelbox/schema/tool_building/base_step_reasoning_tool.py b/libs/labelbox/src/labelbox/schema/tool_building/base_step_reasoning_tool.py
index 4d7ce54c5..7e1f8dbc9 100644
--- a/libs/labelbox/src/labelbox/schema/tool_building/base_step_reasoning_tool.py
+++ b/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
diff --git a/libs/labelbox/src/labelbox/schema/tool_building/classification.py b/libs/labelbox/src/labelbox/schema/tool_building/classification.py
index 9c0c69bea..8c6dc64b4 100644
--- a/libs/labelbox/src/labelbox/schema/tool_building/classification.py
+++ b/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
 
diff --git a/libs/labelbox/src/labelbox/schema/tool_building/fact_checking_tool.py b/libs/labelbox/src/labelbox/schema/tool_building/fact_checking_tool.py
index 440f343cd..2dfb34d10 100644
--- a/libs/labelbox/src/labelbox/schema/tool_building/fact_checking_tool.py
+++ b/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 (
diff --git a/libs/labelbox/src/labelbox/schema/tool_building/prompt_issue_tool.py b/libs/labelbox/src/labelbox/schema/tool_building/prompt_issue_tool.py
index 1a965ff95..9ef56445f 100644
--- a/libs/labelbox/src/labelbox/schema/tool_building/prompt_issue_tool.py
+++ b/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 (
diff --git a/libs/labelbox/src/labelbox/schema/tool_building/step_reasoning_tool.py b/libs/labelbox/src/labelbox/schema/tool_building/step_reasoning_tool.py
index c69c6f22a..12d6ae733 100644
--- a/libs/labelbox/src/labelbox/schema/tool_building/step_reasoning_tool.py
+++ b/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 (
diff --git a/libs/labelbox/src/labelbox/schema/user_group_v2.py b/libs/labelbox/src/labelbox/schema/user_group_v2.py
index a734eb397..c9c291d22 100644
--- a/libs/labelbox/src/labelbox/schema/user_group_v2.py
+++ b/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