update

Author: jtsodapop

labelbox/schema/ontology.py

@@ -98,6 +98,7 @@ class Classification:
         schema_id: (str)
         feature_schema_id: (str)
     """
+
     class Type(Enum):
         TEXT = "text"
         CHECKLIST = "checklist"
@@ -180,6 +181,7 @@ class Tool:
         schema_id: (str)
         feature_schema_id: (str)
     """
+
     class Type(Enum):
         POLYGON = "polygon"
         SEGMENTATION = "superpixel"
@@ -221,8 +223,8 @@ def asdict(self) -> Dict[str, Any]:
         }
 
     def add_classification(self, classification: Classification):
-        if classification.instructions in (c.instructions
-                                           for c in self.classifications):
+        if classification.instructions in (
+                c.instructions for c in self.classifications):
             raise InconsistentOntologyException(
                 f"Duplicate nested classification '{classification.instructions}' "
                 f"for tool '{self.name}'")
@@ -345,9 +347,9 @@ def add_tool(self, tool: Tool) -> Tool:
 
     def add_classification(self,
                            classification: Classification) -> Classification:
-        if classification.instructions in (c.instructions
-                                           for c in self.classifications):
+        if classification.instructions in (
+                c.instructions for c in self.classifications):
             raise InconsistentOntologyException(
                 f"Duplicate classification instructions '{classification.instructions}'. "
             )
-        self.classifications.append(classification)
+        self.classifications.append(classification)

tests/integration/test_ontology.py

@@ -26,39 +26,39 @@
         "classifications": []
     }, {
         "schemaNodeId":
-        None,
+            None,
         "featureSchemaId":
-        None,
+            None,
         "required":
-        False,
+            False,
         "name":
-        "bbox",
+            "bbox",
         "color":
-        "#FF0000",
+            "#FF0000",
         "tool":
-        "rectangle",
+            "rectangle",
         "classifications": [{
             "schemaNodeId":
-            None,
+                None,
             "featureSchemaId":
-            None,
+                None,
             "required":
-            True,
+                True,
             "instructions":
-            "nested classification",
+                "nested classification",
             "name":
-            "nested classification",
+                "nested classification",
             "type":
-            "radio",
+                "radio",
             "options": [{
                 "schemaNodeId":
-                None,
+                    None,
                 "featureSchemaId":
-                None,
+                    None,
                 "label":
-                "first",
+                    "first",
                 "value":
-                "first",
+                    "first",
                 "options": [{
                     "schemaNodeId": None,
                     "featureSchemaId": None,
@@ -111,17 +111,17 @@
     }],
     "classifications": [{
         "schemaNodeId":
-        None,
+            None,
         "featureSchemaId":
-        None,
+            None,
         "required":
-        True,
+            True,
         "instructions":
-        "This is a question.",
+            "This is a question.",
         "name":
-        "This is a question.",
+            "This is a question.",
         "type":
-        "radio",
+            "radio",
         "options": [{
             "schemaNodeId": None,
             "featureSchemaId": None,
@@ -204,8 +204,7 @@ def test_add_ontology_classification() -> None:
 
 def test_tool_add_classification() -> None:
     t = Tool(tool=Tool.Type.SEGMENTATION, name="segmentation")
-    c = Classification(class_type=Classification.Type.TEXT,
-                       instructions="text")
+    c = Classification(class_type=Classification.Type.TEXT, instructions="text")
     t.add_classification(c)
     assert t.classifications == [c]
 
@@ -228,8 +227,7 @@ def test_classification_add_option() -> None:
 
 def test_option_add_option() -> None:
     o = Option(value="option")
-    c = Classification(class_type=Classification.Type.TEXT,
-                       instructions="text")
+    c = Classification(class_type=Classification.Type.TEXT, instructions="text")
     o.add_option(c)
     assert o.options == [c]
 
@@ -245,4 +243,4 @@ def test_ontology_asdict(project) -> None:
 
 def test_from_project_ontology(client, project) -> None:
     o = OntologyBuilder.from_project(project)
-    assert o.asdict() == project.ontology().normalized
+    assert o.asdict() == project.ontology().normalized