clean up imports

Author: Matt Sokoloff

labelbox/data/annotation_types/collection.py

@@ -1,4 +1,3 @@
-from labelbox.data.ontology import get_classifications, get_tools
 import logging
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from typing import Callable, Generator, Iterable, Union
@@ -8,6 +7,7 @@
 
 from labelbox.schema import ontology
 from labelbox.orm.model import Entity
+from ..ontology import get_classifications, get_tools
 from ..generator import PrefetchGenerator
 from .label import Label
 

labelbox/data/annotation_types/label.py

@@ -1,22 +1,18 @@
 from collections import defaultdict
-from labelbox.data.ontology import get_feature_schema_lookup
-
 from typing import Any, Callable, Dict, List, Union
 
-from pydantic import BaseModel
+from pydantic import BaseModel, validator
 
 from labelbox.schema import ontology
 from labelbox.orm.model import Entity
+from ..ontology import get_feature_schema_lookup
 from .classification import ClassificationAnswer
 from .data import VideoData, TextData, RasterData
 from .geometry import Mask
 from .metrics import Metric
 from .annotation import (ClassificationAnnotation, ObjectAnnotation,
                          VideoClassificationAnnotation, VideoObjectAnnotation)
 
-from pydantic import validator
-
-from labelbox.data.annotation_types import geometry
 
 
 class Label(BaseModel):