Start cleaning up unused imports

encord/client.py

@@ -64,11 +64,9 @@
     DatasetUser,
     DatasetUserRole,
     DatasetUsers,
-    DicomSeries,
     Image,
     ImageGroup,
     ImageGroupOCR,
-    Images,
     LongPollingStatus,
     ReEncodeVideoTask,
     Video,
@@ -90,7 +88,6 @@
     ShadowDataState,
 )
 from encord.orm.labeling_algorithm import (
-    BoundingBoxFittingParams,
     LabelingAlgorithm,
     ObjectInterpolationParams,
 )

encord/collection.py

@@ -37,7 +37,7 @@
     UpdateCollectionPayload,
 )
 from encord.orm.collection import ProjectCollection as OrmProjectCollection
-from encord.orm.label_row import LabelRowMetadata, label_row_metadata_dto_to_label_row_metadata
+from encord.orm.label_row import label_row_metadata_dto_to_label_row_metadata
 from encord.storage import StorageItem, StorageItemInaccessible
 
 log = logging.getLogger(__name__)

encord/common/bitmask_operations/__init__.py

@@ -20,6 +20,10 @@
     )
 
 __all__ = [
+    "_mask_to_rle",
+    "_rle_to_mask",
+    "_rle_to_string",
+    "_string_to_rle",
     "deserialise_bitmask",
     "serialise_bitmask",
     "transpose_bytearray",

encord/common/bitmask_operations/bitmask_operations_numpy.py

@@ -1,4 +1,4 @@
-from typing import List, Sequence, Tuple
+from typing import List, Tuple
 
 import numpy as np
 

encord/common/range_manager.py

@@ -1,4 +1,4 @@
-from typing import Iterable, List, Optional, Set, Union, cast
+from typing import Optional, Set, cast
 
 from encord.objects.frames import Frames, Range, Ranges
 

encord/filter_preset.py

@@ -2,7 +2,6 @@
 from typing import Iterator, List, Optional, Union
 from uuid import UUID
 
-from encord.client import EncordClientProject
 from encord.exceptions import (
     AuthorisationError,
     EncordException,

encord/http/__init__.py

@@ -1 +1,5 @@
 from encord.http.constants import RequestsSettings
+
+__all__ = [
+    "RequestsSettings",
+]

encord/http/utils.py

@@ -15,23 +15,16 @@
 from concurrent.futures import ThreadPoolExecutor
 from dataclasses import dataclass
 from pathlib import Path
-from typing import Dict, Iterable, List, Optional, Tuple, Type, Union
+from typing import Dict, Iterable, List, Optional, Union
 
 from tqdm import tqdm
 
 from encord.configs import BaseConfig
 from encord.exceptions import CloudUploadError, EncordException
-from encord.http.querier import Querier, create_new_session
+from encord.http.querier import create_new_session
 from encord.http.v2.api_client import ApiClient
 from encord.http.v2.payloads import Page
 from encord.orm.base_dto import BaseDTO
-from encord.orm.dataset import (
-    Audio,
-    DicomSeries,
-    Images,
-    SignedImagesURL,
-    Video,
-)
 from encord.orm.storage import StorageItemType, UploadSignedUrl
 
 PROGRESS_BAR_FILE_FACTOR = 100

encord/issues/issue_client.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-from dataclasses import dataclass
 from enum import auto
 from typing import List, Literal, Union
 from uuid import UUID

encord/objects/classification.py

@@ -12,7 +12,6 @@
 from __future__ import annotations
 
 from dataclasses import dataclass
-from enum import Enum
 from typing import Any, Dict, List, Optional, Sequence, Type
 
 from encord.common.enum import StringEnum

encord/objects/classification_instance.py

@@ -24,7 +24,6 @@
     NoReturn,
     Optional,
     Sequence,
-    Set,
     Tuple,
     Union,
 )
@@ -42,9 +41,9 @@
     TextAttribute,
     _get_attribute_by_hash,
 )
-from encord.objects.classification import Classification, OntologyClassificationLevel
+from encord.objects.classification import Classification
 from encord.objects.constants import DEFAULT_CONFIDENCE, DEFAULT_MANUAL_ANNOTATION
-from encord.objects.frames import Frames, Ranges, frames_class_to_frames_list, frames_to_ranges
+from encord.objects.frames import Frames, Ranges, frames_class_to_frames_list
 from encord.objects.internal_helpers import (
     _infer_attribute_from_answer,
     _search_child_attributes,

encord/objects/common.py

@@ -5,21 +5,31 @@
 from typing import Any, Dict, List, Union
 
 from encord.common.enum import StringEnum
-from encord.objects.attributes import (  # pylint: disable=unused-import
+from encord.objects.attributes import (
     Attribute,
     ChecklistAttribute,
     RadioAttribute,
     TextAttribute,
 )
-
-# Following imports need to be here for backwards compatibility
-from encord.objects.ontology_element import NestedID  # pylint: disable=unused-import
-from encord.objects.options import (  # pylint: disable=unused-import
+from encord.objects.ontology_element import NestedID
+from encord.objects.options import (
     FlatOption,
     NestableOption,
     Option,
 )
 
+# Following imports need to be here for backwards compatibility
+__all__ = [
+    "Attribute",
+    "ChecklistAttribute",
+    "FlatOption",
+    "NestableOption",
+    "NestedID",
+    "Option",
+    "RadioAttribute",
+    "TextAttribute",
+]
+
 
 class PropertyType(StringEnum):
     RADIO = "radio"

encord/objects/constants.py

@@ -11,9 +11,14 @@
 
 from __future__ import annotations
 
-# for backwards compatibility
 from encord.common.constants import DATETIME_LONG_STRING_FORMAT
 
+# for backwards compatibility
+__all__ = [
+    "DATETIME_LONG_STRING_FORMAT",
+]
+
+
 DEFAULT_CONFIDENCE = 1.0
 DEFAULT_MANUAL_ANNOTATION = True
 AVAILABLE_COLORS = (

encord/objects/html_node.py

@@ -11,8 +11,7 @@
 
 from __future__ import annotations
 
-from dataclasses import dataclass
-from typing import Collection, List, Union, cast
+from typing import List
 
 from encord.orm.base_dto import BaseDTO
 

encord/objects/ontology_labels_impl.py

@@ -71,9 +71,7 @@
     Range,
     Ranges,
     frames_class_to_frames_list,
-    frames_to_ranges,
     ranges_list_to_ranges,
-    ranges_to_list,
 )
 from encord.objects.html_node import HtmlRange
 from encord.objects.metadata import DataGroupMetadata, DICOMSeriesMetadata, DICOMSliceMetadata
@@ -86,7 +84,6 @@
 from encord.orm import storage as orm_storage
 from encord.orm.label_row import (
     AnnotationTaskStatus,
-    LabelRow,
     LabelRowMetadata,
     LabelStatus,
     WorkflowGraphNode,

encord/objects/ontology_object_instance.py

@@ -28,7 +28,6 @@
     Union,
 )
 
-from encord.common.range_manager import RangeManager
 from encord.common.time_parser import parse_datetime
 from encord.constants.enums import DataType
 from encord.exceptions import LabelRowError
@@ -52,7 +51,7 @@
     frames_to_ranges,
     ranges_list_to_ranges,
 )
-from encord.objects.html_node import HtmlRange, HtmlRanges
+from encord.objects.html_node import HtmlRanges
 from encord.objects.internal_helpers import (
     _infer_attribute_from_answer,
     _search_child_attributes,

encord/objects/project.py

@@ -1,2 +1,6 @@
-# Import for backward compatibility
 from encord.orm.project import ProjectDataset
+
+# Export for backward compatibility
+__all__ = [
+    "ProjectDataset",
+]

encord/objects/skeleton_template.py

@@ -11,12 +11,10 @@
 
 from __future__ import annotations
 
-from dataclasses import dataclass
-from typing import Any, List, Optional, Set, Type
+from typing import List, Set
 
 from encord.objects.coordinates import SkeletonCoordinate, SkeletonCoordinates
 from encord.orm.skeleton_template import SkeletonTemplate as SkeletonTemplateORM
-from encord.orm.skeleton_template import SkeletonTemplateCoordinate
 
 
 class SkeletonTemplate(SkeletonTemplateORM):

encord/orm/analytics.py

@@ -1,11 +1,11 @@
 from datetime import datetime
 from enum import Enum, auto
-from typing import List, Literal, Optional
+from typing import List, Optional
 from uuid import UUID
 
 from encord.common.utils import snake_to_camel
 from encord.orm.base_dto import BaseDTO
-from encord.orm.workflow import BaseWorkflowNode, WorkflowNode, WorkflowStageType
+from encord.orm.workflow import BaseWorkflowNode
 from encord.utilities.project_user import ProjectUserRole
 
 

encord/orm/base_dto/__init__.py

@@ -25,3 +25,14 @@
         RootModelDTO,
         dto_validator,
     )
+
+__all__ = [
+    "BaseDTO",
+    "BaseDTOInterface",
+    "BaseDTOWithExtra",
+    "Field",
+    "GenericBaseDTO",
+    "PrivateAttr",
+    "RootModelDTO",
+    "dto_validator",
+]

encord/orm/base_dto/base_dto_pydantic_v1.py

@@ -5,8 +5,8 @@
 
 from pydantic import (
     BaseModel,
-    Field,
-    PrivateAttr,
+    Field,  # noqa: F401 needed for imports elsewhere
+    PrivateAttr,  # noqa: F401 needed for imports elsewhere
     ValidationError,
     root_validator,
     validator,

encord/orm/base_dto/base_dto_pydantic_v2.py

@@ -1,14 +1,12 @@
 from datetime import datetime
-from functools import wraps
 from typing import Any, Callable, Dict, Literal, Type, TypeVar, get_origin
 
 # TODO: invent some dependency version dependent type checking to get rid of this ignore
 from pydantic import (  # type: ignore[attr-defined]
     BaseModel,
     ConfigDict,  # type: ignore[attr-defined]
-    Extra,
-    Field,
-    PrivateAttr,
+    Field,  # noqa: F401 needed for imports elsewhere
+    PrivateAttr,  # noqa: F401 needed for imports elsewhere
     RootModel,  # type: ignore[attr-defined]
     ValidationError,
     field_validator,

encord/orm/base_orm.py

@@ -1,5 +1,4 @@
 import datetime
-import json
 import logging
 from collections import OrderedDict, abc
 

encord/orm/collection.py

@@ -1,7 +1,7 @@
 import uuid
 from datetime import datetime
 from enum import Enum
-from typing import Any, List, Optional, Union
+from typing import List, Optional, Union
 from uuid import UUID
 
 from encord.orm.base_dto import BaseDTO, Field

encord/orm/filter_preset.py

@@ -3,7 +3,7 @@
 from typing import Dict, List, Optional
 from uuid import UUID
 
-from encord.orm.base_dto import BaseDTO, Field, dto_validator
+from encord.orm.base_dto import BaseDTO, Field
 
 
 class GetPresetParams(BaseDTO):

encord/orm/ontology.py

@@ -1,12 +1,11 @@
 from __future__ import annotations
 
 from datetime import datetime
-from enum import IntEnum
 from typing import Dict, Optional
 
 # pylint: disable=unused-import
 from encord.objects.ontology_structure import OntologyStructure
-from encord.orm.base_dto import BaseDTO, dto_validator
+from encord.orm.base_dto import BaseDTO
 from encord.orm.formatter import Formatter
 from encord.utilities.ontology_user import OntologyUserRole
 

encord/orm/workflow.py

@@ -2,8 +2,6 @@
 from typing import Any, List, Union
 from uuid import UUID
 
-from typing_extensions import Annotated
-
 from encord.orm.base_dto import BaseDTO, dto_validator
 
 

encord/storage.py

@@ -17,7 +17,7 @@
 from datetime import datetime
 from math import ceil
 from pathlib import Path
-from typing import Any, Collection, Dict, Iterable, List, Literal, Optional, Sequence, TextIO, Union
+from typing import Any, Collection, Dict, Iterable, List, Optional, Sequence, TextIO, Union
 from uuid import UUID
 
 import requests

encord/user_client.py

@@ -32,7 +32,7 @@
 from encord.common.deprecated import deprecated
 from encord.common.time_parser import parse_datetime, parse_datetime_optional
 from encord.configs import ENCORD_DOMAIN, BearerConfig, SshConfig, UserConfig, get_env_ssh_key
-from encord.constants.string_constants import TYPE_DATASET, TYPE_ONTOLOGY, TYPE_PROJECT
+from encord.constants.string_constants import TYPE_DATASET, TYPE_PROJECT
 from encord.dataset import Dataset
 from encord.filter_preset import FilterPreset
 from encord.http.constants import DEFAULT_REQUESTS_SETTINGS, RequestsSettings
@@ -62,12 +62,9 @@
     DatasetInfo,
     DatasetsWithUserRolesListParams,
     DatasetsWithUserRolesListResponse,
-    DicomDeidentifyTask,
-    Images,
     StorageLocation,
     dataset_user_role_str_enum_to_int_enum,
 )
-from encord.orm.dataset import Dataset as OrmDataset
 from encord.orm.deidentification import (
     DicomDeIdGetResultLongPollingStatus,
     DicomDeIdGetResultParams,
@@ -79,7 +76,6 @@
 )
 from encord.orm.group import Group as OrmGroup
 from encord.orm.ontology import CreateOrUpdateOntologyPayload
-from encord.orm.ontology import Ontology as OrmOntology
 from encord.orm.project import (
     BenchmarkQaWorkflowSettings,
     CvatExportType,
@@ -108,7 +104,7 @@
     Issues,
     LocalImport,
 )
-from encord.utilities.ontology_user import OntologiesFilterParams, OntologyUserRole, OntologyWithUserRole
+from encord.utilities.ontology_user import OntologiesFilterParams, OntologyWithUserRole
 from encord.utilities.project_user import ProjectUserRole
 
 CVAT_LONG_POLLING_RESPONSE_RETRY_N = 3