yapf

Author: Matt Sokoloff

labelbox/data/annotation_types/geometry/rectangle.py

@@ -27,7 +27,7 @@ def geometry(self) -> geojson.geometry.Geometry:
         ]])
 
     def raster(self, height: int, width: int,
-               color = (255, 255, 255)) -> np.ndarray:
+               color=(255, 255, 255)) -> np.ndarray:
         """
         Draw the rectangle onto a 3d mask
 

labelbox/data/annotation_types/label.py

@@ -14,7 +14,6 @@
                          VideoClassificationAnnotation, VideoObjectAnnotation)
 
 
-
 class Label(BaseModel):
     data: Union[VideoData, RasterData, TextData]
     annotations: List[Union[ClassificationAnnotation, ObjectAnnotation,

labelbox/data/ontology.py

@@ -1,20 +1,9 @@
 from typing import Dict, List, Tuple
 
 from labelbox.schema import ontology
-from .annotation_types import (
-    Text,
-    Dropdown,
-    Checklist,
-    Radio,
-    ClassificationAnnotation,
-    ObjectAnnotation,
-    Mask,
-    Point,
-    Line,
-    Polygon,
-    Rectangle,
-    TextEntity
-)
+from .annotation_types import (Text, Dropdown, Checklist, Radio,
+                               ClassificationAnnotation, ObjectAnnotation, Mask,
+                               Point, Line, Polygon, Rectangle, TextEntity)
 
 
 def get_feature_schema_lookup(

setup.py

@@ -23,7 +23,9 @@
         "backoff==1.10.0",
         "backports-datetime-fromisoformat==1.0.0; python_version < '3.7.0'",
         "dataclasses==0.7; python_version < '3.7.0'", "ndjson==0.3.1",
-        "requests>=2.22.0", "google-api-core>=1.22.1", "pydantic>=1.8,<2.0", "shapely", "geojson", "numpy", "rasterio", "PILLOW", "opencv-python", "typeguard", "tqdm"
+        "requests>=2.22.0", "google-api-core>=1.22.1", "pydantic>=1.8,<2.0",
+        "shapely", "geojson", "numpy", "rasterio", "PILLOW", "opencv-python",
+        "typeguard", "tqdm"
     ],
     classifiers=[
         'Development Status :: 3 - Alpha',

tests/data/annotation_types/classification/test_classification.py

@@ -1,14 +1,9 @@
 import pytest
 from pydantic import ValidationError
 
-from labelbox.data.annotation_types import (
-    Checklist,
-    ClassificationAnswer,
-    Dropdown,
-    Radio,
-    Text,
-    ClassificationAnnotation
-)
+from labelbox.data.annotation_types import (Checklist, ClassificationAnswer,
+                                            Dropdown, Radio, Text,
+                                            ClassificationAnnotation)
 
 
 def test_classification_answer():

tests/data/annotation_types/data/test_text.py

@@ -3,6 +3,7 @@
 
 from labelbox.data.annotation_types import TextData
 
+
 def test_validate_schema():
     with pytest.raises(ValidationError):
         data = TextData()

tests/data/annotation_types/geometry/test_mask.py

@@ -49,10 +49,10 @@ def test_mask():
     gt1 = Rectangle(start=Point(x=0, y=0),
                     end=Point(x=10, y=10)).raster(height=raster1.shape[0],
                                                   width=raster1.shape[1],
-                                                  color=(255,255,255))
+                                                  color=(255, 255, 255))
     gt2 = Rectangle(start=Point(x=20, y=20),
                     end=Point(x=30, y=30)).raster(height=raster2.shape[0],
                                                   width=raster2.shape[1],
-                                                  color=(0,255,255))
+                                                  color=(0, 255, 255))
     assert (raster1 == gt1).all()
     assert (raster2 == gt2).all()

tests/data/annotation_types/test_annotation.py

@@ -1,18 +1,11 @@
-
 import pytest
 from pydantic import ValidationError
 
-from labelbox.data.annotation_types import (
-    Text,
-    Point,
-    Line,
-    ClassificationAnnotation,
-    ObjectAnnotation,
-    VideoClassificationAnnotation,
-    VideoObjectAnnotation,
-    TextEntity
-)
-
+from labelbox.data.annotation_types import (Text, Point, Line,
+                                            ClassificationAnnotation,
+                                            ObjectAnnotation,
+                                            VideoClassificationAnnotation,
+                                            VideoObjectAnnotation, TextEntity)
 
 
 def test_annotation():

tests/data/annotation_types/test_collection.py

@@ -4,16 +4,9 @@
 import numpy as np
 import pytest
 
-from labelbox.data.annotation_types import (
-    LabelList,
-    LabelGenerator,
-    ObjectAnnotation,
-    RasterData,
-    Line,
-    Mask,
-    Point,
-    Label
-)
+from labelbox.data.annotation_types import (LabelList, LabelGenerator,
+                                            ObjectAnnotation, RasterData, Line,
+                                            Mask, Point, Label)
 from labelbox import OntologyBuilder, Tool
 
 

tests/data/annotation_types/test_label.py

@@ -1,17 +1,10 @@
 import numpy as np
 
 from labelbox import OntologyBuilder, Tool, Classification as OClassification, Option
-from labelbox.data.annotation_types import (
-    ClassificationAnswer,
-    Radio,
-    Text,
-    ClassificationAnnotation,
-    ObjectAnnotation,
-    Point,
-    Line,
-    RasterData,
-    Label
-)
+from labelbox.data.annotation_types import (ClassificationAnswer, Radio, Text,
+                                            ClassificationAnnotation,
+                                            ObjectAnnotation, Point, Line,
+                                            RasterData, Label)
 
 
 def test_schema_assignment_geometry():