Update typings

Author: Val Brodsky

libs/labelbox/src/labelbox/annotated_types.py

@@ -1,6 +1,5 @@
-from typing_extensions import Annotated
+from typing import Annotated
 
 from pydantic import Field
 
-
 Cuid = Annotated[str, Field(min_length=25, max_length=25)]

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

@@ -1,6 +1,6 @@
 from abc import ABC
 from io import BytesIO
-from typing import Callable, Optional, Union
+from typing import Callable, Literal, Optional, Union
 
 import numpy as np
 import requests
@@ -9,9 +9,6 @@
 from PIL import Image
 from pydantic import BaseModel, ConfigDict, model_validator
 from requests.exceptions import ConnectTimeout
-from typing_extensions import Literal
-
-from pydantic import BaseModel, model_validator, ConfigDict
 
 from ..types import TypedArray
 

libs/labelbox/src/labelbox/data/annotation_types/metrics/scalar.py

@@ -1,11 +1,10 @@
-from typing import Dict, Optional, Union
-from typing_extensions import Annotated
 from enum import Enum
+from typing import Annotated, Dict, Optional, Union
 
 from pydantic import field_validator
 from pydantic.types import confloat
 
-from .base import ConfidenceValue, BaseMetric
+from .base import BaseMetric, ConfidenceValue
 
 ScalarMetricValue = Annotated[float, confloat(ge=0, le=100_000_000)]
 ScalarMetricConfidenceValue = Dict[ConfidenceValue, ScalarMetricValue]

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

@@ -1,11 +1,9 @@
 import sys
-from typing import Generic, TypeVar, Any
+from typing import Annotated, Any, Generic, TypeVar
 
-from typing_extensions import Annotated
-from packaging import version
 import numpy as np
-
-from pydantic import StringConstraints, Field, ConfigDict
+from packaging import version
+from pydantic import ConfigDict, Field, StringConstraints
 from pydantic_core import core_schema
 
 DType = TypeVar("DType")

libs/labelbox/src/labelbox/data/metrics/group.py

@@ -16,10 +16,10 @@
 try:
     from typing import Literal
 except ImportError:
-    from typing_extensions import Literal
+    from typing import Literal
 
+from ..annotation_types import ClassificationAnnotation, Label, ObjectAnnotation
 from ..annotation_types.feature import FeatureSchema
-from ..annotation_types import ObjectAnnotation, ClassificationAnnotation, Label
 
 
 def get_identifying_key(

libs/labelbox/src/labelbox/project_validation.py

@@ -1,7 +1,6 @@
-from typing import Optional, Set
+from typing import Annotated, Optional, Set
 
 from pydantic import BaseModel, ConfigDict, Field, model_validator
-from typing_extensions import Annotated
 
 from labelbox.schema.media_type import MediaType
 from labelbox.schema.ontology_kind import EditorTaskType

libs/labelbox/src/labelbox/schema/data_row_metadata.py

@@ -1,35 +1,34 @@
 # type: ignore
-from datetime import datetime
+import warnings
 from copy import deepcopy
+from datetime import datetime
 from enum import Enum
 from itertools import chain
-import warnings
-
 from typing import (
+    Annotated,
+    Any,
+    Callable,
+    Dict,
+    Generator,
     List,
     Optional,
-    Dict,
-    Union,
-    Callable,
     Type,
-    Any,
-    Generator,
+    Union,
     overload,
 )
-from typing_extensions import Annotated
 
-from labelbox.schema.identifiables import DataRowIdentifiers, UniqueIds
-from labelbox.schema.identifiable import UniqueId, GlobalKey
 from pydantic import (
     BaseModel,
+    BeforeValidator,
+    ConfigDict,
     Field,
     StringConstraints,
     conlist,
-    ConfigDict,
     model_serializer,
-    BeforeValidator,
 )
 
+from labelbox.schema.identifiable import GlobalKey, UniqueId
+from labelbox.schema.identifiables import DataRowIdentifiers, UniqueIds
 from labelbox.schema.ontology import SchemaId
 from labelbox.utils import (
     _CamelCaseMixin,

libs/labelbox/src/labelbox/schema/export_filters.py

@@ -1,7 +1,5 @@
 from datetime import datetime, timezone
-from typing import Collection, Dict, List, Optional, Tuple, TypedDict
-
-from labelbox.typing_imports import Literal
+from typing import Collection, Dict, List, Literal, Optional, Tuple, TypedDict
 
 SEARCH_LIMIT_PER_EXPORT_V2 = 2_000
 ISO_8061_FORMAT = "%Y-%m-%dT%H:%M:%S%z"

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

@@ -5,11 +5,10 @@
 import warnings
 from dataclasses import dataclass, field
 from enum import Enum
-from typing import Any, Dict, List, Optional, Type, Union
+from typing import Annotated, Any, Dict, List, Optional, Type, Union
 
 from lbox.exceptions import InconsistentOntologyException
 from pydantic import StringConstraints
-from typing_extensions import Annotated
 
 from labelbox.orm.db_object import DbObject
 from labelbox.orm.model import Field, Relationship

libs/labelbox/src/labelbox/schema/project_overview.py

@@ -1,5 +1,5 @@
-from typing import Dict, List
-from typing_extensions import TypedDict
+from typing import Dict, List, TypedDict
+
 from pydantic import BaseModel