Remove typing_extensions

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/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,8 @@
-import sys
-from typing import Generic, TypeVar, Any
+from typing import Any, Generic, TypeVar
 
-from typing_extensions import Annotated
-from packaging import version
 import numpy as np
-
-from pydantic import StringConstraints, Field
+from packaging import version
+from pydantic import Field
 
 DType = TypeVar("DType")
 DShape = TypeVar("DShape")

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

@@ -1,34 +1,33 @@
 # 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,
+    ConfigDict,
     Field,
     StringConstraints,
     conlist,
-    ConfigDict,
     model_serializer,
 )
 
+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,
@@ -417,7 +416,7 @@ def update_enum_option(
         schema = self._validate_custom_schema_by_name(name)
         if schema.kind != DataRowMetadataKind.enum:
             raise ValueError(
-                f"Updating Enum option is only supported for Enum metadata schema"
+                "Updating Enum option is only supported for Enum metadata schema"
             )
         valid_options: List[str] = [o.name for o in schema.options]
 
@@ -803,13 +802,13 @@ def _convert_metadata_field(metadata_field):
             if isinstance(metadata_field, DataRowMetadataField):
                 return metadata_field
             elif isinstance(metadata_field, dict):
-                if not "value" in metadata_field:
+                if "value" not in metadata_field:
                     raise ValueError(
                         f"Custom metadata field '{metadata_field}' must have a 'value' key"
                     )
                 if (
-                    not "schema_id" in metadata_field
-                    and not "name" in metadata_field
+                    "schema_id" not in metadata_field
+                    and "name" not in metadata_field
                 ):
                     raise ValueError(
                         f"Custom metadata field '{metadata_field}' must have either 'schema_id' or 'name' key"

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

@@ -1,8 +1,6 @@
 
 from datetime import datetime, timezone
-from typing import Collection, Dict, List, Literal, Optional, Tuple
-
-from typing_extensions import TypedDict
+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/export_params.py

@@ -1,6 +1,4 @@
-from typing import List, Optional
-
-from typing_extensions import TypedDict
+from typing import List, Optional, TypedDict
 
 from labelbox.schema.media_type import MediaType
 

libs/labelbox/src/labelbox/schema/labeling_service.py

@@ -1,13 +1,10 @@
 from datetime import datetime
 from typing import Any
-from typing_extensions import Annotated
-
-from pydantic import BaseModel, Field
 
 from labelbox.exceptions import ResourceNotFoundError
-from labelbox.utils import _CamelCaseMixin
 from labelbox.schema.labeling_service_dashboard import LabelingServiceDashboard
 from labelbox.schema.labeling_service_status import LabelingServiceStatus
+from labelbox.utils import _CamelCaseMixin
 
 from ..annotated_types import Cuid
 

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

@@ -1,17 +1,17 @@
 # type: ignore
 
 import colorsys
+import json
+import warnings
 from dataclasses import dataclass, field
 from enum import Enum
-from typing import Any, Dict, List, Optional, Union, Type
-from typing_extensions import Annotated
-import warnings
+from typing import Annotated, Any, Dict, List, Optional, Type, Union
+
+from pydantic import StringConstraints
 
 from labelbox.exceptions import InconsistentOntologyException
 from labelbox.orm.db_object import DbObject
 from labelbox.orm.model import Field, Relationship
-import json
-from pydantic import StringConstraints
 
 FeatureSchemaId: Type[str] = Annotated[
     str, StringConstraints(min_length=25, max_length=25)

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
 
 

libs/labelbox/src/labelbox/schema/search_filters.py

@@ -1,11 +1,9 @@
 import datetime
 from enum import Enum
-from typing import List, Union
-from pydantic import PlainSerializer, BaseModel, Field
+from typing import Annotated, List, Union
 
-from typing_extensions import Annotated
+from pydantic import BaseModel, Field, PlainSerializer, field_validator
 
-from pydantic import BaseModel, Field, field_validator
 from labelbox.schema.labeling_service_status import LabelingServiceStatus
 from labelbox.utils import format_iso_datetime