diff --git a/libs/labelbox/pyproject.toml b/libs/labelbox/pyproject.toml index a47466004..0bf40c07c 100644 --- a/libs/labelbox/pyproject.toml +++ b/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] diff --git a/libs/labelbox/src/labelbox/client.py b/libs/labelbox/src/labelbox/client.py index bcf29665e..4041c0a1e 100644 --- a/libs/labelbox/src/labelbox/client.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/data/annotation_types/collection.py b/libs/labelbox/src/labelbox/data/annotation_types/collection.py index 42d2a1184..03c8f5343 100644 --- a/libs/labelbox/src/labelbox/data/annotation_types/collection.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/data/annotation_types/data/tiled_image.py b/libs/labelbox/src/labelbox/data/annotation_types/data/tiled_image.py index adb8db549..7edc5ce3b 100644 --- a/libs/labelbox/src/labelbox/data/annotation_types/data/tiled_image.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/data/annotation_types/feature.py b/libs/labelbox/src/labelbox/data/annotation_types/feature.py index 621a53a11..af9cbb16f 100644 --- a/libs/labelbox/src/labelbox/data/annotation_types/feature.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/data/annotation_types/label.py b/libs/labelbox/src/labelbox/data/annotation_types/label.py index 2f835b23c..d13fb8f20 100644 --- a/libs/labelbox/src/labelbox/data/annotation_types/label.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/data/annotation_types/metrics/base.py b/libs/labelbox/src/labelbox/data/annotation_types/metrics/base.py index 0a4773a41..981122eb6 100644 --- a/libs/labelbox/src/labelbox/data/annotation_types/metrics/base.py +++ b/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) diff --git a/libs/labelbox/src/labelbox/data/annotation_types/metrics/confusion_matrix.py b/libs/labelbox/src/labelbox/data/annotation_types/metrics/confusion_matrix.py index 30e2d2ed4..15d8e9a0b 100644 --- a/libs/labelbox/src/labelbox/data/annotation_types/metrics/confusion_matrix.py +++ b/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) diff --git a/libs/labelbox/src/labelbox/data/annotation_types/video.py b/libs/labelbox/src/labelbox/data/annotation_types/video.py index 5a93704c8..14a692bae 100644 --- a/libs/labelbox/src/labelbox/data/annotation_types/video.py +++ b/libs/labelbox/src/labelbox/data/annotation_types/video.py @@ -20,7 +20,6 @@ model_validator, BaseModel, field_validator, - model_serializer, Field, ConfigDict, AliasChoices, diff --git a/libs/labelbox/src/labelbox/data/metrics/confusion_matrix/confusion_matrix.py b/libs/labelbox/src/labelbox/data/metrics/confusion_matrix/confusion_matrix.py index 6d817b105..c1a524729 100644 --- a/libs/labelbox/src/labelbox/data/metrics/confusion_matrix/confusion_matrix.py +++ b/libs/labelbox/src/labelbox/data/metrics/confusion_matrix/confusion_matrix.py @@ -1,10 +1,7 @@ # 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, ) @@ -12,7 +9,6 @@ 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( diff --git a/libs/labelbox/src/labelbox/data/metrics/group.py b/libs/labelbox/src/labelbox/data/metrics/group.py index 9c4104c29..63718aad4 100644 --- a/libs/labelbox/src/labelbox/data/metrics/group.py +++ b/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, ) diff --git a/libs/labelbox/src/labelbox/data/serialization/ndjson/base.py b/libs/labelbox/src/labelbox/data/serialization/ndjson/base.py index d8d8cd36f..4b9cc0489 100644 --- a/libs/labelbox/src/labelbox/data/serialization/ndjson/base.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/data/serialization/ndjson/converter.py b/libs/labelbox/src/labelbox/data/serialization/ndjson/converter.py index 8176d7862..4d3c160b5 100644 --- a/libs/labelbox/src/labelbox/data/serialization/ndjson/converter.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/data/serialization/ndjson/label.py b/libs/labelbox/src/labelbox/data/serialization/ndjson/label.py index ffaefb4d7..af938e511 100644 --- a/libs/labelbox/src/labelbox/data/serialization/ndjson/label.py +++ b/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, diff --git a/libs/labelbox/src/labelbox/data/serialization/ndjson/mmc.py b/libs/labelbox/src/labelbox/data/serialization/ndjson/mmc.py index b2dcfb5b4..44c36747d 100644 --- a/libs/labelbox/src/labelbox/data/serialization/ndjson/mmc.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/data/serialization/ndjson/objects.py b/libs/labelbox/src/labelbox/data/serialization/ndjson/objects.py index 1bcba7a89..28f3a45c8 100644 --- a/libs/labelbox/src/labelbox/data/serialization/ndjson/objects.py +++ b/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, ) diff --git a/libs/labelbox/src/labelbox/data/serialization/ndjson/relationship.py b/libs/labelbox/src/labelbox/data/serialization/ndjson/relationship.py index d558ac244..f692bae41 100644 --- a/libs/labelbox/src/labelbox/data/serialization/ndjson/relationship.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/schema/asset_attachment.py b/libs/labelbox/src/labelbox/schema/asset_attachment.py index 9a56dbb72..69804f64c 100644 --- a/libs/labelbox/src/labelbox/schema/asset_attachment.py +++ b/libs/labelbox/src/labelbox/schema/asset_attachment.py @@ -1,4 +1,3 @@ -import warnings from enum import Enum from typing import Dict, Optional diff --git a/libs/labelbox/src/labelbox/schema/catalog.py b/libs/labelbox/src/labelbox/schema/catalog.py index 8d9646779..a077ab374 100644 --- a/libs/labelbox/src/labelbox/schema/catalog.py +++ b/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 ( diff --git a/libs/labelbox/src/labelbox/schema/create_batches_task.py b/libs/labelbox/src/labelbox/schema/create_batches_task.py index 25ff80917..b8770f8e8 100644 --- a/libs/labelbox/src/labelbox/schema/create_batches_task.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/schema/data_row.py b/libs/labelbox/src/labelbox/schema/data_row.py index cb0e99b22..e5e95832d 100644 --- a/libs/labelbox/src/labelbox/schema/data_row.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/schema/data_row_metadata.py b/libs/labelbox/src/labelbox/schema/data_row_metadata.py index cb45ef57f..58111828f 100644 --- a/libs/labelbox/src/labelbox/schema/data_row_metadata.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/schema/export_task.py b/libs/labelbox/src/labelbox/schema/export_task.py index 7e78fc3e9..9673e42ef 100644 --- a/libs/labelbox/src/labelbox/schema/export_task.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/schema/foundry/app.py b/libs/labelbox/src/labelbox/schema/foundry/app.py index 2886dec15..afe51b1c6 100644 --- a/libs/labelbox/src/labelbox/schema/foundry/app.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/schema/internal/data_row_uploader.py b/libs/labelbox/src/labelbox/schema/internal/data_row_uploader.py index 817a02561..5159d52ff 100644 --- a/libs/labelbox/src/labelbox/schema/internal/data_row_uploader.py +++ b/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, diff --git a/libs/labelbox/src/labelbox/schema/model_run.py b/libs/labelbox/src/labelbox/schema/model_run.py index dcdbdf0e8..94ffb29dd 100644 --- a/libs/labelbox/src/labelbox/schema/model_run.py +++ b/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, diff --git a/libs/labelbox/src/labelbox/schema/project.py b/libs/labelbox/src/labelbox/schema/project.py index a6f2dfe28..94833938e 100644 --- a/libs/labelbox/src/labelbox/schema/project.py +++ b/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, diff --git a/libs/labelbox/src/labelbox/schema/project_resource_tag.py b/libs/labelbox/src/labelbox/schema/project_resource_tag.py index 18ca94860..33d5f3468 100644 --- a/libs/labelbox/src/labelbox/schema/project_resource_tag.py +++ b/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): diff --git a/libs/labelbox/src/labelbox/schema/resource_tag.py b/libs/labelbox/src/labelbox/schema/resource_tag.py index 8c0559486..26481f052 100644 --- a/libs/labelbox/src/labelbox/schema/resource_tag.py +++ b/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): diff --git a/libs/labelbox/src/labelbox/schema/role.py b/libs/labelbox/src/labelbox/schema/role.py index 47cd753e9..d22e2a78e 100644 --- a/libs/labelbox/src/labelbox/schema/role.py +++ b/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: diff --git a/libs/labelbox/src/labelbox/schema/serialization.py b/libs/labelbox/src/labelbox/schema/serialization.py index ca5537fd9..305af1f4c 100644 --- a/libs/labelbox/src/labelbox/schema/serialization.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/schema/slice.py b/libs/labelbox/src/labelbox/schema/slice.py index a640ebc1d..b41d7162b 100644 --- a/libs/labelbox/src/labelbox/schema/slice.py +++ b/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 ( diff --git a/libs/labelbox/src/labelbox/schema/webhook.py b/libs/labelbox/src/labelbox/schema/webhook.py index 0eebe157e..54f6eace9 100644 --- a/libs/labelbox/src/labelbox/schema/webhook.py +++ b/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 diff --git a/libs/labelbox/src/labelbox/utils.py b/libs/labelbox/src/labelbox/utils.py index dcf51be82..aacfb9a2f 100644 --- a/libs/labelbox/src/labelbox/utils.py +++ b/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" diff --git a/libs/labelbox/tests/conftest.py b/libs/labelbox/tests/conftest.py index 3e9f0b491..a07d52c4d 100644 --- a/libs/labelbox/tests/conftest.py +++ b/libs/labelbox/tests/conftest.py @@ -19,7 +19,6 @@ Client, DataRow, Dataset, - LabelingFrontend, MediaType, OntologyBuilder, Option, diff --git a/libs/labelbox/tests/data/annotation_import/test_annotation_import_limit.py b/libs/labelbox/tests/data/annotation_import/test_annotation_import_limit.py index dec20fbb5..ae5df24e4 100644 --- a/libs/labelbox/tests/data/annotation_import/test_annotation_import_limit.py +++ b/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 diff --git a/libs/labelbox/tests/data/annotation_import/test_generic_data_types.py b/libs/labelbox/tests/data/annotation_import/test_generic_data_types.py index 921e98c9d..e7ff7c684 100644 --- a/libs/labelbox/tests/data/annotation_import/test_generic_data_types.py +++ b/libs/labelbox/tests/data/annotation_import/test_generic_data_types.py @@ -6,11 +6,6 @@ import labelbox as lb from labelbox import Client, OntologyKind, Project -from labelbox.data.annotation_types import Label -from labelbox.data.annotation_types.data.generic_data_row_data import ( - GenericDataRowData, -) -from labelbox.data.serialization.ndjson.converter import NDJsonConverter from labelbox.schema.annotation_import import AnnotationImportState from labelbox.schema.media_type import MediaType diff --git a/libs/labelbox/tests/data/annotation_import/test_mea_prediction_import.py b/libs/labelbox/tests/data/annotation_import/test_mea_prediction_import.py index 5f47975ad..f309cf188 100644 --- a/libs/labelbox/tests/data/annotation_import/test_mea_prediction_import.py +++ b/libs/labelbox/tests/data/annotation_import/test_mea_prediction_import.py @@ -1,11 +1,6 @@ import uuid from labelbox import parser from labelbox.data.annotation_types.annotation import ObjectAnnotation -from labelbox.data.annotation_types.classification.classification import ( - ClassificationAnnotation, - ClassificationAnswer, - Radio, -) from labelbox.data.annotation_types.data.generic_data_row_data import ( GenericDataRowData, ) diff --git a/libs/labelbox/tests/data/annotation_import/test_send_to_annotate_mea.py b/libs/labelbox/tests/data/annotation_import/test_send_to_annotate_mea.py index 4bcd4dcef..73a621608 100644 --- a/libs/labelbox/tests/data/annotation_import/test_send_to_annotate_mea.py +++ b/libs/labelbox/tests/data/annotation_import/test_send_to_annotate_mea.py @@ -1,5 +1,3 @@ -import pytest - from labelbox import UniqueIds, OntologyBuilder from labelbox.schema.conflict_resolution_strategy import ( ConflictResolutionStrategy, diff --git a/libs/labelbox/tests/data/annotation_types/data/test_raster.py b/libs/labelbox/tests/data/annotation_types/data/test_raster.py index 209419aed..7e97eb960 100644 --- a/libs/labelbox/tests/data/annotation_types/data/test_raster.py +++ b/libs/labelbox/tests/data/annotation_types/data/test_raster.py @@ -1,4 +1,3 @@ -import urllib.request from io import BytesIO import numpy as np diff --git a/libs/labelbox/tests/data/annotation_types/test_collection.py b/libs/labelbox/tests/data/annotation_types/test_collection.py index e0fa7bd53..9e4c8cff5 100644 --- a/libs/labelbox/tests/data/annotation_types/test_collection.py +++ b/libs/labelbox/tests/data/annotation_types/test_collection.py @@ -7,14 +7,11 @@ from labelbox.data.annotation_types import ( LabelGenerator, ObjectAnnotation, - Line, Mask, - Point, Label, GenericDataRowData, MaskData, ) -from labelbox import OntologyBuilder, Tool @pytest.fixture diff --git a/libs/labelbox/tests/data/annotation_types/test_label.py b/libs/labelbox/tests/data/annotation_types/test_label.py index 9cd992b0c..d2dd2b634 100644 --- a/libs/labelbox/tests/data/annotation_types/test_label.py +++ b/libs/labelbox/tests/data/annotation_types/test_label.py @@ -1,18 +1,7 @@ -from pydantic import ValidationError import numpy as np import labelbox.types as lb_types -from labelbox import ( - OntologyBuilder, - Tool, - Classification as OClassification, - Option, -) from labelbox.data.annotation_types import ( - ClassificationAnswer, - Radio, - Text, - ClassificationAnnotation, PromptText, ObjectAnnotation, Point, diff --git a/libs/labelbox/tests/data/export/conftest.py b/libs/labelbox/tests/data/export/conftest.py index 4a59b6966..4d54e3cbc 100644 --- a/libs/labelbox/tests/data/export/conftest.py +++ b/libs/labelbox/tests/data/export/conftest.py @@ -5,7 +5,6 @@ from labelbox import Client, MediaType from labelbox.schema.annotation_import import AnnotationImportState, LabelImport -from labelbox.schema.labeling_frontend import LabelingFrontend from labelbox.schema.media_type import MediaType diff --git a/libs/labelbox/tests/data/export/streamable/test_export_dataset_streamable.py b/libs/labelbox/tests/data/export/streamable/test_export_dataset_streamable.py index 0d34a40b1..1de58c92b 100644 --- a/libs/labelbox/tests/data/export/streamable/test_export_dataset_streamable.py +++ b/libs/labelbox/tests/data/export/streamable/test_export_dataset_streamable.py @@ -1,5 +1,3 @@ -import json - import pytest from labelbox import ExportTask, StreamType diff --git a/libs/labelbox/tests/data/export/streamable/test_export_embeddings_streamable.py b/libs/labelbox/tests/data/export/streamable/test_export_embeddings_streamable.py index 803b5994a..e18f68f05 100644 --- a/libs/labelbox/tests/data/export/streamable/test_export_embeddings_streamable.py +++ b/libs/labelbox/tests/data/export/streamable/test_export_embeddings_streamable.py @@ -1,4 +1,3 @@ -import json import random from labelbox import StreamType diff --git a/libs/labelbox/tests/data/export/streamable/test_export_model_run_streamable.py b/libs/labelbox/tests/data/export/streamable/test_export_model_run_streamable.py index 7a583198b..d859656a5 100644 --- a/libs/labelbox/tests/data/export/streamable/test_export_model_run_streamable.py +++ b/libs/labelbox/tests/data/export/streamable/test_export_model_run_streamable.py @@ -1,6 +1,3 @@ -import json -import time - from labelbox import ExportTask, StreamType diff --git a/libs/labelbox/tests/data/export/streamable/test_export_project_streamable.py b/libs/labelbox/tests/data/export/streamable/test_export_project_streamable.py index 597c529aa..4cb00aaa6 100644 --- a/libs/labelbox/tests/data/export/streamable/test_export_project_streamable.py +++ b/libs/labelbox/tests/data/export/streamable/test_export_project_streamable.py @@ -1,5 +1,3 @@ -import json - import pytest import uuid from typing import Tuple diff --git a/libs/labelbox/tests/data/export/streamable/test_export_video_streamable.py b/libs/labelbox/tests/data/export/streamable/test_export_video_streamable.py index da99414ee..eec56ff51 100644 --- a/libs/labelbox/tests/data/export/streamable/test_export_video_streamable.py +++ b/libs/labelbox/tests/data/export/streamable/test_export_video_streamable.py @@ -1,5 +1,3 @@ -import json - import pytest import labelbox as lb diff --git a/libs/labelbox/tests/data/metrics/iou/data_row/conftest.py b/libs/labelbox/tests/data/metrics/iou/data_row/conftest.py index 6614cecf4..6c03c2839 100644 --- a/libs/labelbox/tests/data/metrics/iou/data_row/conftest.py +++ b/libs/labelbox/tests/data/metrics/iou/data_row/conftest.py @@ -1,8 +1,6 @@ -from io import BytesIO from types import SimpleNamespace import pytest import numpy as np -from PIL import Image import base64 diff --git a/libs/labelbox/tests/data/metrics/iou/feature/conftest.py b/libs/labelbox/tests/data/metrics/iou/feature/conftest.py index c3b2a28e3..3ee50994f 100644 --- a/libs/labelbox/tests/data/metrics/iou/feature/conftest.py +++ b/libs/labelbox/tests/data/metrics/iou/feature/conftest.py @@ -3,7 +3,6 @@ import pytest from labelbox.data.annotation_types import ( - ClassificationAnnotation, ObjectAnnotation, ) from labelbox.data.annotation_types import Polygon, Point diff --git a/libs/labelbox/tests/integration/conftest.py b/libs/labelbox/tests/integration/conftest.py index a0c2499a5..c248bf67e 100644 --- a/libs/labelbox/tests/integration/conftest.py +++ b/libs/labelbox/tests/integration/conftest.py @@ -1,24 +1,17 @@ -import json import os -import re import sys import time -import uuid from collections import defaultdict from datetime import datetime, timezone -from enum import Enum from itertools import islice -from types import SimpleNamespace -from typing import List, Tuple, Type +from typing import Type import pytest from labelbox import ( Classification, Client, - DataRow, Dataset, - LabelingFrontend, MediaType, OntologyBuilder, Option, @@ -26,13 +19,7 @@ ResponseOption, Tool, ) -from labelbox.orm import query -from labelbox.pagination import PaginatedCollection -from labelbox.schema.annotation_import import LabelImport -from labelbox.schema.catalog import Catalog from labelbox.schema.data_row import DataRowMetadataField -from labelbox.schema.enums import AnnotationImportState -from labelbox.schema.invite import Invite from labelbox.schema.ontology_kind import OntologyKind from labelbox.schema.user import User diff --git a/libs/labelbox/tests/integration/test_batches.py b/libs/labelbox/tests/integration/test_batches.py index cabae4053..7ea04a826 100644 --- a/libs/labelbox/tests/integration/test_batches.py +++ b/libs/labelbox/tests/integration/test_batches.py @@ -1,6 +1,5 @@ from typing import List -import pytest from labelbox import Project, Dataset diff --git a/libs/labelbox/tests/integration/test_chat_evaluation_ontology_project.py b/libs/labelbox/tests/integration/test_chat_evaluation_ontology_project.py index 796cd9859..2c02b77ac 100644 --- a/libs/labelbox/tests/integration/test_chat_evaluation_ontology_project.py +++ b/libs/labelbox/tests/integration/test_chat_evaluation_ontology_project.py @@ -1,5 +1,3 @@ -from unittest.mock import patch - import pytest from labelbox import MediaType diff --git a/libs/labelbox/tests/integration/test_client_errors.py b/libs/labelbox/tests/integration/test_client_errors.py index 38022d1d2..b5444eadf 100644 --- a/libs/labelbox/tests/integration/test_client_errors.py +++ b/libs/labelbox/tests/integration/test_client_errors.py @@ -7,7 +7,6 @@ from google.api_core.exceptions import RetryError import labelbox.client -from labelbox import Project, User from labelbox.schema.media_type import MediaType diff --git a/libs/labelbox/tests/integration/test_data_upload.py b/libs/labelbox/tests/integration/test_data_upload.py index e10dac943..c4f7d206e 100644 --- a/libs/labelbox/tests/integration/test_data_upload.py +++ b/libs/labelbox/tests/integration/test_data_upload.py @@ -1,4 +1,3 @@ -import pytest import requests diff --git a/libs/labelbox/tests/integration/test_legacy_project.py b/libs/labelbox/tests/integration/test_legacy_project.py index 3e652f333..b2d479ec9 100644 --- a/libs/labelbox/tests/integration/test_legacy_project.py +++ b/libs/labelbox/tests/integration/test_legacy_project.py @@ -1,5 +1,3 @@ -from os import name - import pytest from pydantic import ValidationError diff --git a/libs/labelbox/tests/integration/test_pagination.py b/libs/labelbox/tests/integration/test_pagination.py index 1d500c9fa..9f31c4131 100644 --- a/libs/labelbox/tests/integration/test_pagination.py +++ b/libs/labelbox/tests/integration/test_pagination.py @@ -1,5 +1,4 @@ from copy import copy -import time import pytest diff --git a/libs/labelbox/tests/integration/test_project.py b/libs/labelbox/tests/integration/test_project.py index ea995c6f6..6a9825cfc 100644 --- a/libs/labelbox/tests/integration/test_project.py +++ b/libs/labelbox/tests/integration/test_project.py @@ -7,8 +7,7 @@ import requests from lbox.exceptions import InvalidQueryError -from labelbox import Dataset, LabelingFrontend, Project -from labelbox.schema import media_type +from labelbox import Dataset, Project from labelbox.schema.media_type import MediaType from labelbox.schema.quality_mode import QualityMode diff --git a/libs/labelbox/tests/integration/test_project_setup.py b/libs/labelbox/tests/integration/test_project_setup.py index a6fba03e0..31cf5e066 100644 --- a/libs/labelbox/tests/integration/test_project_setup.py +++ b/libs/labelbox/tests/integration/test_project_setup.py @@ -1,11 +1,7 @@ -import json import time from datetime import datetime, timedelta, timezone import pytest -from lbox.exceptions import InvalidQueryError - -from labelbox import LabelingFrontend def simple_ontology(): diff --git a/libs/labelbox/tests/unit/test_client.py b/libs/labelbox/tests/unit/test_client.py index 9a97591f3..718e47f65 100644 --- a/libs/labelbox/tests/unit/test_client.py +++ b/libs/labelbox/tests/unit/test_client.py @@ -1,7 +1,3 @@ -import os - -from unittest.mock import patch - from labelbox.client import Client diff --git a/libs/labelbox/tests/unit/test_project.py b/libs/labelbox/tests/unit/test_project.py index 1bc6fa840..42559e2dc 100644 --- a/libs/labelbox/tests/unit/test_project.py +++ b/libs/labelbox/tests/unit/test_project.py @@ -1,5 +1,5 @@ import pytest -from unittest.mock import MagicMock, patch +from unittest.mock import MagicMock from labelbox.schema.project import Project from labelbox.schema.ontology_kind import EditorTaskType diff --git a/libs/labelbox/tests/unit/test_unit_delete_batch_data_row_metadata.py b/libs/labelbox/tests/unit/test_unit_delete_batch_data_row_metadata.py index cd6eadd79..b0100adf6 100644 --- a/libs/labelbox/tests/unit/test_unit_delete_batch_data_row_metadata.py +++ b/libs/labelbox/tests/unit/test_unit_delete_batch_data_row_metadata.py @@ -1,5 +1,3 @@ -from re import U - from labelbox.schema.data_row_metadata import _DeleteBatchDataRowMetadata from labelbox.schema.identifiable import GlobalKey, UniqueId diff --git a/libs/labelbox/tests/unit/test_unit_descriptor_file_creator.py b/libs/labelbox/tests/unit/test_unit_descriptor_file_creator.py index 621317ddd..6e92cfb6f 100644 --- a/libs/labelbox/tests/unit/test_unit_descriptor_file_creator.py +++ b/libs/labelbox/tests/unit/test_unit_descriptor_file_creator.py @@ -1,7 +1,6 @@ import json -from unittest.mock import MagicMock, Mock -import pytest +from unittest.mock import MagicMock from labelbox.schema.internal.descriptor_file_creator import ( DescriptorFileCreator, diff --git a/libs/labelbox/tests/unit/test_unit_query.py b/libs/labelbox/tests/unit/test_unit_query.py index 83bfeff8a..4924c598f 100644 --- a/libs/labelbox/tests/unit/test_unit_query.py +++ b/libs/labelbox/tests/unit/test_unit_query.py @@ -1,5 +1,3 @@ -import pytest - from labelbox import Project, Dataset from labelbox.orm import query from labelbox.orm.comparison import Comparison, LogicalExpression diff --git a/libs/labelbox/tests/unit/test_unit_search_filters.py b/libs/labelbox/tests/unit/test_unit_search_filters.py index b2230bb7f..4276c6742 100644 --- a/libs/labelbox/tests/unit/test_unit_search_filters.py +++ b/libs/labelbox/tests/unit/test_unit_search_filters.py @@ -9,7 +9,6 @@ DateRangeValue, DateValue, IdOperator, - OperationType, OrganizationFilter, ProjectStageFilter, SharedWithOrganizationFilter,