Vb/support pydantic2 plt 145 (#1412)

Author: vbrodsky

labelbox/data/annotation_types/base_annotation.py

@@ -1,15 +1,15 @@
 import abc
 from uuid import UUID
 from typing import Any, Dict, Optional
-from pydantic import PrivateAttr
+from labelbox import pydantic_compat
 
 from .feature import FeatureSchema
 
 
 class BaseAnnotation(FeatureSchema, abc.ABC):
     """ Base annotation class. Shouldn't be directly instantiated
     """
-    _uuid: Optional[UUID] = PrivateAttr()
+    _uuid: Optional[UUID] = pydantic_compat.PrivateAttr()
     extra: Dict[str, Any] = {}
 
     def __init__(self, **data):

labelbox/data/annotation_types/classification/classification.py

@@ -9,12 +9,12 @@
 except:
     from typing_extensions import Literal
 
-from pydantic import BaseModel, validator
+from labelbox import pydantic_compat
 from ..feature import FeatureSchema
 
 
 # TODO: Replace when pydantic adds support for unions that don't coerce types
-class _TempName(ConfidenceMixin, BaseModel):
+class _TempName(ConfidenceMixin, pydantic_compat.BaseModel):
     name: str
 
     def dict(self, *args, **kwargs):
@@ -47,7 +47,7 @@ def dict(self, *args, **kwargs) -> Dict[str, str]:
         return res
 
 
-class Radio(ConfidenceMixin, CustomMetricsMixin, BaseModel):
+class Radio(ConfidenceMixin, CustomMetricsMixin, pydantic_compat.BaseModel):
     """ A classification with only one selected option allowed
 
     >>> Radio(answer = ClassificationAnswer(name = "dog"))
@@ -66,7 +66,7 @@ class Checklist(_TempName):
     answer: List[ClassificationAnswer]
 
 
-class Text(ConfidenceMixin, CustomMetricsMixin, BaseModel):
+class Text(ConfidenceMixin, CustomMetricsMixin, pydantic_compat.BaseModel):
     """ Free form text
 
     >>> Text(answer = "some text answer")

labelbox/data/annotation_types/data/base_data.py

@@ -1,10 +1,10 @@
 from abc import ABC
 from typing import Optional, Dict, List, Any
 
-from pydantic import BaseModel
+from labelbox import pydantic_compat
 
 
-class BaseData(BaseModel, ABC):
+class BaseData(pydantic_compat.BaseModel, ABC):
     """
     Base class for objects representing data.
     This class shouldn't directly be used

labelbox/data/annotation_types/data/raster.py

@@ -5,18 +5,17 @@
 
 from PIL import Image
 from google.api_core import retry
-from pydantic import BaseModel
-from pydantic import root_validator
 from requests.exceptions import ConnectTimeout
 import requests
 import numpy as np
 
+from labelbox import pydantic_compat
 from labelbox.exceptions import InternalServerError
 from .base_data import BaseData
 from ..types import TypedArray
 
 
-class RasterData(BaseModel, ABC):
+class RasterData(pydantic_compat.BaseModel, ABC):
     """Represents an image or segmentation mask.
     """
     im_bytes: Optional[bytes] = None
@@ -156,7 +155,7 @@ def create_url(self, signer: Callable[[bytes], str]) -> str:
                 "One of url, im_bytes, file_path, arr must not be None.")
         return self.url
 
-    @root_validator()
+    @pydantic_compat.root_validator()
     def validate_args(cls, values):
         file_path = values.get("file_path")
         im_bytes = values.get("im_bytes")

labelbox/data/annotation_types/data/text.py

@@ -3,8 +3,8 @@
 import requests
 from requests.exceptions import ConnectTimeout
 from google.api_core import retry
-from pydantic import root_validator
 
+from labelbox import pydantic_compat
 from labelbox.exceptions import InternalServerError
 from labelbox.typing_imports import Literal
 from labelbox.utils import _NoCoercionMixin
@@ -90,7 +90,7 @@ def create_url(self, signer: Callable[[bytes], str]) -> None:
                 "One of url, im_bytes, file_path, numpy must not be None.")
         return self.url
 
-    @root_validator
+    @pydantic_compat.root_validator
     def validate_date(cls, values):
         file_path = values.get("file_path")
         text = values.get("text")

labelbox/data/annotation_types/data/tiled_image.py

@@ -12,8 +12,7 @@
 from PIL import Image
 from pyproj import Transformer
 from pygeotile.point import Point as PygeoPoint
-from pydantic import BaseModel, validator
-from pydantic.class_validators import root_validator
+from labelbox import pydantic_compat
 
 from labelbox.data.annotation_types import Rectangle, Point, Line, Polygon
 from .base_data import BaseData
@@ -41,7 +40,7 @@ class EPSG(Enum):
     EPSG3857 = 3857
 
 
-class TiledBounds(BaseModel):
+class TiledBounds(pydantic_compat.BaseModel):
     """ Bounds for a tiled image asset related to the relevant epsg.
 
     Bounds should be Point objects.
@@ -55,7 +54,7 @@ class TiledBounds(BaseModel):
     epsg: EPSG
     bounds: List[Point]
 
-    @validator('bounds')
+    @pydantic_compat.validator('bounds')
     def validate_bounds_not_equal(cls, bounds):
         first_bound = bounds[0]
         second_bound = bounds[1]
@@ -67,7 +66,7 @@ def validate_bounds_not_equal(cls, bounds):
         return bounds
 
     #validate bounds are within lat,lng range if they are EPSG4326
-    @root_validator
+    @pydantic_compat.root_validator
     def validate_bounds_lat_lng(cls, values):
         epsg = values.get('epsg')
         bounds = values.get('bounds')
@@ -83,7 +82,7 @@ def validate_bounds_lat_lng(cls, values):
         return values
 
 
-class TileLayer(BaseModel):
+class TileLayer(pydantic_compat.BaseModel):
     """ Url that contains the tile layer. Must be in the format:
 
     https://c.tile.openstreetmap.org/{z}/{x}/{y}.png
@@ -99,7 +98,7 @@ class TileLayer(BaseModel):
     def asdict(self) -> Dict[str, str]:
         return {"tileLayerUrl": self.url, "name": self.name}
 
-    @validator('url')
+    @pydantic_compat.validator('url')
     def validate_url(cls, url):
         xyz_format = "/{z}/{x}/{y}"
         if xyz_format not in url:
@@ -344,7 +343,7 @@ def _validate_num_tiles(self, xstart: float, ystart: float, xend: float,
                              f"Max allowed tiles are {max_tiles}"
                              f"Increase max tiles or reduce zoom level.")
 
-    @validator('zoom_levels')
+    @pydantic_compat.validator('zoom_levels')
     def validate_zoom_levels(cls, zoom_levels):
         if zoom_levels[0] > zoom_levels[1]:
             raise ValueError(
@@ -353,7 +352,7 @@ def validate_zoom_levels(cls, zoom_levels):
         return zoom_levels
 
 
-class EPSGTransformer(BaseModel):
+class EPSGTransformer(pydantic_compat.BaseModel):
     """Transformer class between different EPSG's. Useful when wanting to project
     in different formats.
     """

labelbox/data/annotation_types/data/video.py

@@ -8,11 +8,12 @@
 import cv2
 import numpy as np
 from google.api_core import retry
-from pydantic import root_validator
 
 from .base_data import BaseData
 from ..types import TypedArray
 
+from labelbox import pydantic_compat
+
 logger = logging.getLogger(__name__)
 
 
@@ -147,7 +148,7 @@ def frames_to_video(self,
         out.release()
         return file_path
 
-    @root_validator
+    @pydantic_compat.root_validator
     def validate_data(cls, values):
         file_path = values.get("file_path")
         url = values.get("url")

labelbox/data/annotation_types/feature.py

@@ -1,11 +1,11 @@
 from typing import Optional
 
-from pydantic import BaseModel, root_validator
+from labelbox import pydantic_compat
 
 from .types import Cuid
 
 
-class FeatureSchema(BaseModel):
+class FeatureSchema(pydantic_compat.BaseModel):
     """
     Class that represents a feature schema.
     Could be a annotation, a subclass, or an option.
@@ -14,7 +14,7 @@ class FeatureSchema(BaseModel):
     name: Optional[str] = None
     feature_schema_id: Optional[Cuid] = None
 
-    @root_validator
+    @pydantic_compat.root_validator
     def must_set_one(cls, values):
         if values['feature_schema_id'] is None and values['name'] is None:
             raise ValueError(

labelbox/data/annotation_types/geometry/geometry.py

@@ -3,12 +3,12 @@
 
 import geojson
 import numpy as np
-from pydantic import BaseModel
+from labelbox import pydantic_compat
 
 from shapely import geometry as geom
 
 
-class Geometry(BaseModel, ABC):
+class Geometry(pydantic_compat.BaseModel, ABC):
     """Abstract base class for geometry objects
     """
     extra: Dict[str, Any] = {}

labelbox/data/annotation_types/geometry/line.py

@@ -3,13 +3,14 @@
 import geojson
 import numpy as np
 import cv2
-from pydantic import validator
 
 from shapely.geometry import LineString as SLineString
 
 from .point import Point
 from .geometry import Geometry
 
+from labelbox import pydantic_compat
+
 
 class Line(Geometry):
     """Line annotation
@@ -64,7 +65,7 @@ def draw(self,
                              color=color,
                              thickness=thickness)
 
-    @validator('points')
+    @pydantic_compat.validator('points')
     def is_geom_valid(cls, points):
         if len(points) < 2:
             raise ValueError(