[PLT-1587] Removed all unused imports (#1876)

Author: Gabefire

libs/labelbox/pyproject.toml

@@ -74,7 +74,7 @@ dev-dependencies = [
 line-length = 80
 
 [tool.ruff.lint]
-ignore = ["F", "E722"]
+ignore = ["F841", "E722", "F811", "F402", "F601", "F403", "F821", "F541"]
 exclude = ["**/__init__.py"]
 
 [tool.rye.scripts]

libs/labelbox/src/labelbox/client.py

@@ -8,9 +8,8 @@
 import urllib.parse
 import warnings
 from collections import defaultdict
-from datetime import datetime, timezone
 from types import MappingProxyType
-from typing import Any, Callable, Dict, List, Optional, Set, Union, overload
+from typing import Any, Callable, Dict, List, Optional, Set, Union
 
 import requests
 import requests.exceptions
@@ -67,10 +66,6 @@
 from labelbox.schema.organization import Organization
 from labelbox.schema.project import Project
 from labelbox.schema.quality_mode import (
-    BENCHMARK_AUTO_AUDIT_NUMBER_OF_LABELS,
-    BENCHMARK_AUTO_AUDIT_PERCENTAGE,
-    CONSENSUS_AUTO_AUDIT_NUMBER_OF_LABELS,
-    CONSENSUS_AUTO_AUDIT_PERCENTAGE,
     QualityMode,
 )
 from labelbox.schema.role import Role

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

@@ -1,9 +1,6 @@
 import logging
-import warnings
 from typing import Callable, Generator, Iterable, Union
 
-from labelbox.orm.model import Entity
-from labelbox.schema import ontology
 
 from ..generator import PrefetchGenerator
 from .label import Label

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

@@ -1,4 +1,3 @@
-from functools import lru_cache
 import math
 import logging
 from enum import Enum

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

@@ -1,5 +1,5 @@
 from typing import Optional
-from pydantic import BaseModel, model_validator, model_serializer
+from pydantic import BaseModel, model_validator
 
 from ...annotated_types import Cuid
 

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

@@ -1,10 +1,8 @@
 from collections import defaultdict
 from typing import Any, Callable, Dict, List, Union, Optional, get_args
-import warnings
 
 import labelbox
 from labelbox.data.annotation_types.data import GenericDataRowData, MaskData
-from labelbox.schema import ontology
 
 from ...annotated_types import Cuid
 from .annotation import ClassificationAnnotation, ObjectAnnotation
@@ -16,7 +14,6 @@
 from .video import VideoClassificationAnnotation
 from .video import VideoObjectAnnotation, VideoMaskAnnotation
 from .mmc import MessageEvaluationTaskAnnotation
-from ..ontology import get_feature_schema_lookup
 from pydantic import BaseModel, field_validator
 
 

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

@@ -6,9 +6,7 @@
     BaseModel,
     model_serializer,
     field_validator,
-    error_wrappers,
 )
-from pydantic_core import ValidationError, InitErrorDetails
 
 ConfidenceValue = confloat(ge=0, le=1)
 

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

@@ -1,10 +1,9 @@
 from enum import Enum
 from typing import Optional, Tuple, Dict, Union
 
-from pydantic import conint, Field
+from pydantic import conint
 
 from .base import ConfidenceValue, BaseMetric
-from typing import Literal
 
 Count = conint(ge=0, le=1e10)
 

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

@@ -20,7 +20,6 @@
     model_validator,
     BaseModel,
     field_validator,
-    model_serializer,
     Field,
     ConfigDict,
     AliasChoices,

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

@@ -1,18 +1,14 @@
 # type: ignore
-from collections import defaultdict
-from labelbox.data.annotation_types import feature
 from labelbox.data.annotation_types.metrics import ConfusionMatrixMetric
-from typing import List, Optional, Union
+from typing import List, Union
 from ...annotation_types import (
-    Label,
     ObjectAnnotation,
     ClassificationAnnotation,
 )
 
 from ..group import get_feature_pairs
 from .calculation import confusion_matrix
 from .calculation import feature_confusion_matrix
-import numpy as np
 
 
 def confusion_matrix_metric(

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

@@ -8,7 +8,6 @@
 from labelbox.data.annotation_types.annotation import ClassificationAnnotation
 from labelbox.data.annotation_types.classification.classification import (
     Checklist,
-    ClassificationAnswer,
     Radio,
     Text,
 )

libs/labelbox/src/labelbox/data/serialization/ndjson/base.py

@@ -2,9 +2,8 @@
 from uuid import uuid4
 
 from labelbox.utils import _CamelCaseMixin, is_exactly_one_set
-from pydantic import model_validator, ConfigDict, BaseModel, Field
+from pydantic import model_validator, Field
 from uuid import uuid4
-import threading
 
 from ....annotated_types import Cuid
 

libs/labelbox/src/labelbox/data/serialization/ndjson/converter.py

@@ -2,7 +2,7 @@
 import logging
 import uuid
 from collections import defaultdict, deque
-from typing import Any, Deque, Dict, Generator, Iterable, List, Set, Union
+from typing import Any, Deque, Dict, Generator, List, Set, Union
 
 from labelbox.data.annotation_types.annotation import ObjectAnnotation
 from labelbox.data.annotation_types.classification.classification import (
@@ -14,7 +14,7 @@
 from labelbox.data.annotation_types.metrics.scalar import ScalarMetric
 from labelbox.data.annotation_types.video import VideoMaskAnnotation
 
-from ...annotation_types.collection import LabelCollection, LabelGenerator
+from ...annotation_types.collection import LabelCollection
 from ...annotation_types.relationship import RelationshipAnnotation
 from ...annotation_types.mmc import MessageEvaluationTaskAnnotation
 from .label import NDLabel

libs/labelbox/src/labelbox/data/serialization/ndjson/label.py

@@ -9,14 +9,12 @@
 )
 from ...annotation_types.relationship import RelationshipAnnotation
 from ...annotation_types.video import (
-    DICOMObjectAnnotation,
     VideoClassificationAnnotation,
 )
 from ...annotation_types.video import VideoObjectAnnotation, VideoMaskAnnotation
 from ...annotation_types.collection import LabelCollection, LabelGenerator
 from ...annotation_types.data.generic_data_row_data import GenericDataRowData
 from ...annotation_types.label import Label
-from ...annotation_types.ner import TextEntity, ConversationEntity
 from ...annotation_types.metrics import ScalarMetric, ConfusionMatrixMetric
 from ...annotation_types.llm_prompt_response.prompt import (
     PromptClassificationAnnotation,
@@ -44,9 +42,7 @@
 from .mmc import NDMessageTask
 from .relationship import NDRelationship
 from .base import DataRow
-from pydantic import BaseModel, ValidationError
-from pydantic_core import PydanticUndefined
-from contextlib import suppress
+from pydantic import BaseModel
 
 AnnotationType = Union[
     NDObjectType,

libs/labelbox/src/labelbox/data/serialization/ndjson/mmc.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Optional, Union
+from typing import Union
 
 from labelbox.utils import _CamelCaseMixin
 

libs/labelbox/src/labelbox/data/serialization/ndjson/objects.py

@@ -20,7 +20,6 @@
 import numpy as np
 
 from PIL import Image
-from labelbox.data.annotation_types import feature
 
 from labelbox.data.annotation_types.data import GenericDataRowData
 
@@ -49,7 +48,6 @@
     MaskInstance,
 )
 from .classification import (
-    NDClassification,
     NDSubclassification,
     NDSubclassificationType,
 )

libs/labelbox/src/labelbox/data/serialization/ndjson/relationship.py

@@ -1,4 +1,3 @@
-from typing import Union
 from pydantic import BaseModel
 from .base import NDAnnotation, DataRow
 from ...annotation_types.data import GenericDataRowData

libs/labelbox/src/labelbox/schema/asset_attachment.py

@@ -1,4 +1,3 @@
-import warnings
 from enum import Enum
 from typing import Dict, Optional
 

libs/labelbox/src/labelbox/schema/catalog.py

@@ -1,6 +1,5 @@
 from typing import Any, Dict, List, Optional, Tuple, Union
 import warnings
-from labelbox.orm.db_object import experimental
 from labelbox.schema.export_filters import CatalogExportFilters, build_filters
 
 from labelbox.schema.export_params import (

libs/labelbox/src/labelbox/schema/create_batches_task.py

@@ -4,7 +4,6 @@
 from labelbox.orm.model import Entity
 
 if TYPE_CHECKING:
-    from labelbox import User
 
     def lru_cache() -> Callable[..., Callable[..., Dict[str, Any]]]:
         pass

libs/labelbox/src/labelbox/schema/data_row.py

@@ -1,6 +1,6 @@
 import logging
 from enum import Enum
-from typing import TYPE_CHECKING, List, Optional, Tuple, Union, Any
+from typing import TYPE_CHECKING, List, Optional, Tuple, Union
 import json
 import warnings
 
@@ -9,7 +9,6 @@
     DbObject,
     Updateable,
     BulkDeletable,
-    experimental,
 )
 from labelbox.orm.model import Entity, Field, Relationship
 from labelbox.schema.asset_attachment import AttachmentType

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

@@ -1,5 +1,4 @@
 # type: ignore
-import warnings
 from copy import deepcopy
 from datetime import datetime
 from enum import Enum

libs/labelbox/src/labelbox/schema/export_task.py

@@ -1,7 +1,6 @@
 import json
 import os
 import tempfile
-import warnings
 from abc import ABC, abstractmethod
 from dataclasses import dataclass
 from enum import Enum

libs/labelbox/src/labelbox/schema/foundry/app.py

@@ -1,6 +1,5 @@
 from typing import Any, Dict, Optional
-from pydantic import BaseModel, ConfigDict, AliasGenerator
-from pydantic.alias_generators import to_camel, to_snake
+from pydantic import ConfigDict
 from labelbox.utils import _CamelCaseMixin
 
 

libs/labelbox/src/labelbox/schema/internal/data_row_uploader.py

@@ -1,11 +1,7 @@
-from concurrent.futures import ThreadPoolExecutor, as_completed
-
 from typing import List
 
 from labelbox.schema.internal.data_row_upsert_item import (
     DataRowItemBase,
-    DataRowUpsertItem,
-    DataRowCreateItem,
 )
 from labelbox.schema.internal.descriptor_file_creator import (
     DescriptorFileCreator,

libs/labelbox/src/labelbox/schema/model_run.py

@@ -16,9 +16,7 @@
     Any,
 )
 
-import requests
 
-from labelbox import parser
 from labelbox.orm.db_object import DbObject, experimental
 from labelbox.orm.model import Field, Relationship, Entity
 from labelbox.orm.query import results_query_part
@@ -28,7 +26,7 @@
 )
 from labelbox.schema.export_params import ModelRunExportParams
 from labelbox.schema.export_task import ExportTask
-from labelbox.schema.identifiables import UniqueIds, GlobalKeys, DataRowIds
+from labelbox.schema.identifiables import GlobalKeys, DataRowIds
 from labelbox.schema.send_to_annotate_params import (
     SendToAnnotateFromModelParams,
     build_destination_task_queue_input,

libs/labelbox/src/labelbox/schema/project.py

@@ -38,12 +38,9 @@
 )
 from labelbox.schema.export_params import ProjectExportParams
 from labelbox.schema.export_task import ExportTask
-from labelbox.schema.id_type import IdType
-from labelbox.schema.identifiable import DataRowIdentifier, GlobalKey, UniqueId
+from labelbox.schema.identifiable import DataRowIdentifier
 from labelbox.schema.identifiables import (
     DataRowIdentifiers,
-    GlobalKeys,
-    UniqueIds,
 )
 from labelbox.schema.labeling_service import (
     LabelingService,

libs/labelbox/src/labelbox/schema/project_resource_tag.py

@@ -1,5 +1,5 @@
 from labelbox.orm.db_object import DbObject, Updateable
-from labelbox.orm.model import Field, Relationship
+from labelbox.orm.model import Field
 
 
 class ProjectResourceTag(DbObject, Updateable):

libs/labelbox/src/labelbox/schema/resource_tag.py

@@ -1,5 +1,5 @@
 from labelbox.orm.db_object import DbObject, Updateable
-from labelbox.orm.model import Field, Relationship
+from labelbox.orm.model import Field
 
 
 class ResourceTag(DbObject, Updateable):

libs/labelbox/src/labelbox/schema/role.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 from typing import Dict, Optional, TYPE_CHECKING
 
-from labelbox.orm.model import Field, Entity
+from labelbox.orm.model import Field
 from labelbox.orm.db_object import DbObject
 
 if TYPE_CHECKING:

libs/labelbox/src/labelbox/schema/serialization.py

@@ -1,4 +1,4 @@
-from typing import cast, Any, Dict, Generator, List, TYPE_CHECKING, Union
+from typing import cast, Any, Dict, List, TYPE_CHECKING, Union
 
 if TYPE_CHECKING:
     from labelbox.types import Label

libs/labelbox/src/labelbox/schema/slice.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 from typing import Optional, Tuple, Union
 import warnings
-from labelbox.orm.db_object import DbObject, experimental
+from labelbox.orm.db_object import DbObject
 from labelbox.orm.model import Field
 from labelbox.pagination import PaginatedCollection
 from labelbox.schema.export_params import (

libs/labelbox/src/labelbox/schema/webhook.py

@@ -1,6 +1,6 @@
 import logging
 from enum import Enum
-from typing import Iterable, List
+from typing import Iterable
 
 from labelbox.orm import query
 from labelbox.orm.db_object import DbObject, Updateable

libs/labelbox/src/labelbox/utils.py

@@ -10,10 +10,8 @@
     BaseModel,
     ConfigDict,
     model_serializer,
-    AliasGenerator,
-    AliasChoices,
 )
-from pydantic.alias_generators import to_camel, to_pascal
+from pydantic.alias_generators import to_camel
 
 UPPERCASE_COMPONENTS = ["uri", "rgb"]
 ISO_DATETIME_FORMAT = "%Y-%m-%dT%H:%M:%SZ"

libs/labelbox/tests/conftest.py

@@ -19,7 +19,6 @@
     Client,
     DataRow,
     Dataset,
-    LabelingFrontend,
     MediaType,
     OntologyBuilder,
     Option,

libs/labelbox/tests/data/annotation_import/test_annotation_import_limit.py

@@ -1,8 +1,6 @@
 import itertools
-import uuid
 from labelbox.schema.annotation_import import (
     AnnotationImport,
-    MALPredictionImport,
 )
 from labelbox.schema.media_type import MediaType
 import pytest