Import exception fron lbox

Author: Val Brodsky

libs/labelbox/src/labelbox/data/annotation_types/data/raster.py

@@ -1,18 +1,18 @@
 from abc import ABC
 from io import BytesIO
 from typing import Callable, Optional, Union
-from typing_extensions import Literal
 
-from PIL import Image
+import numpy as np
+import requests
 from google.api_core import retry
+from lbox.exceptions import InternalServerError
+from PIL import Image
+from pydantic import BaseModel, ConfigDict, model_validator
 from requests.exceptions import ConnectTimeout
-import requests
-import numpy as np
+from typing_extensions import Literal
 
-from pydantic import BaseModel, model_validator, ConfigDict
-from labelbox.exceptions import InternalServerError
-from .base_data import BaseData
 from ..types import TypedArray
+from .base_data import BaseData
 
 
 class RasterData(BaseModel, ABC):

libs/labelbox/src/labelbox/data/annotation_types/data/text.py

@@ -1,13 +1,14 @@
 from typing import Callable, Optional
 
 import requests
-from requests.exceptions import ConnectTimeout
 from google.api_core import retry
-
+from lbox.exceptions import InternalServerError
 from pydantic import ConfigDict, model_validator
-from labelbox.exceptions import InternalServerError
+from requests.exceptions import ConnectTimeout
+
 from labelbox.typing_imports import Literal
 from labelbox.utils import _NoCoercionMixin
+
 from .base_data import BaseData
 
 

libs/labelbox/src/labelbox/data/mixins.py

@@ -1,13 +1,10 @@
-from typing import Optional, List
+from typing import List, Optional
 
-from pydantic import BaseModel, field_validator, model_serializer
-
-from labelbox.exceptions import (
+from lbox.exceptions import (
     ConfidenceNotSupportedException,
     CustomMetricsNotSupportedException,
 )
-
-from warnings import warn
+from pydantic import BaseModel, field_validator
 
 
 class ConfidenceMixin(BaseModel):

libs/labelbox/src/labelbox/exceptions.py

@@ -1,17 +1,17 @@
-from dataclasses import dataclass
+import json
+import logging
 from datetime import datetime, timezone
 from functools import wraps
-import logging
-import json
 
-from labelbox import utils
-from labelbox.exceptions import (
-    InvalidQueryError,
+from lbox.exceptions import (
     InvalidAttributeError,
+    InvalidQueryError,
     OperationNotSupportedException,
 )
+
+from labelbox import utils
 from labelbox.orm import query
-from labelbox.orm.model import Field, Relationship, Entity
+from labelbox.orm.model import Entity, Field, Relationship
 from labelbox.pagination import PaginatedCollection
 
 logger = logging.getLogger(__name__)

libs/labelbox/src/labelbox/orm/db_object.py

@@ -1,10 +1,11 @@
 from dataclasses import dataclass
 from enum import Enum, auto
-from typing import Dict, List, Union, Any, Type, TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, Dict, List, Type, Union
+
+from lbox.exceptions import InvalidAttributeError
 
 import labelbox
 from labelbox import utils
-from labelbox.exceptions import InvalidAttributeError
 from labelbox.orm.comparison import Comparison
 
 """ Defines Field, Relationship and Entity. These classes are building

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

@@ -1,14 +1,15 @@
 from itertools import chain
 from typing import Any, Dict
 
-from labelbox import utils
-from labelbox.exceptions import (
-    InvalidQueryError,
+from lbox.exceptions import (
     InvalidAttributeError,
+    InvalidQueryError,
     MalformedQueryException,
 )
-from labelbox.orm.comparison import LogicalExpression, Comparison
-from labelbox.orm.model import Field, Relationship, Entity
+
+from labelbox import utils
+from labelbox.orm.comparison import Comparison, LogicalExpression
+from labelbox.orm.model import Entity, Field, Relationship
 
 """ Common query creation functionality. """