complete ontology management

Author: Matt Sokoloff

examples/basics/ontologies.ipynb

@@ -51,7 +51,7 @@
    "id": "18d7cb3b",
    "metadata": {},
    "source": [
-    "### Create Ontology From Normalized JSON\n",
+    "### Create Ontology From Normalized Data\n",
     "* Users can create ontologies from a json definition of the ontology\n",
     "* See below `OntologyBuilder` section for more details on constructing the normalized ontology"
    ]
@@ -65,14 +65,14 @@
    "source": [
     "# This will automatically create new root schema node\n",
     "ontology_name = \"sdk-ontology\"\n",
-    "root_schema_node_cat_normalized_json = {\n",
+    "root_schema_node_cat_normalized = {\n",
     "     'tool': 'polygon', \n",
     "     'name': 'cat', \n",
     "     'color': 'black'\n",
     "}\n",
     "\n",
-    "ontology_normalized_json = {\"tools\" : [root_schema_node_cat_normalized_json], \"classifications\" : []}\n",
-    "ontology = client.create_ontology(name = ontology_name, normalized_json = ontology_normalized_json)\n",
+    "ontology_normalized_json = {\"tools\" : [root_schema_node_cat_normalized], \"classifications\" : []}\n",
+    "ontology = client.create_ontology(name = ontology_name, normalized = ontology_normalized_json)\n",
     "print(ontology)"
    ]
   },
@@ -94,8 +94,9 @@
    "outputs": [],
    "source": [
     "# First create the root schema node\n",
-    "root_schema_node_cat = client.create_root_schema_node(root_schema_node_cat_normalized_json)\n",
+    "root_schema_node_cat = client.create_root_schema_node(root_schema_node_cat_normalized)\n",
     "# When we create the ontology it will not re-create the schema node\n",
+    "print(root_schema_node_cat.uid)\n",
     "ontology = client.create_ontology_from_root_schema_nodes(ontology_name, [root_schema_node_cat.uid])"
    ]
   },
@@ -118,13 +119,13 @@
    "outputs": [],
    "source": [
     "# Create new dog schema id \n",
-    "root_schema_node_dog_normalized_json = {\n",
+    "root_schema_node_dog_normalized = {\n",
     "     'tool': 'polygon', \n",
     "     'name': 'dog', \n",
     "     'color': 'black', \n",
     "     'classifications': [], \n",
     "}\n",
-    "root_schema_node_dog = client.create_root_schema_node(root_schema_node_cat_normalized_json)\n",
+    "root_schema_node_dog = client.create_root_schema_node(root_schema_node_dog_normalized)\n",
     "# The cat is shared between this new ontology and the one we created previously \n",
     "# (ie. the cat root schema node will not be re-created)\n",
     "ontology = client.create_ontology(ontology_name, [root_schema_node_cat.uid, root_schema_node_dog.uid])"

labelbox/client.py

@@ -788,7 +788,7 @@ def create_ontology_from_root_schema_nodes(self, name,
         """
         tools, classifications = [], []
         for schema_node_id in root_schema_node_ids:
-            schema_node = self.get_root_schema_node(root_schema_node_ids)
+            schema_node = self.get_root_schema_node(schema_node_id)
             tool = schema_node.normalized['tool']
             try:
                 Tool.Type(tool)
@@ -801,7 +801,9 @@ def create_ontology_from_root_schema_nodes(self, name,
                     raise ValueError(
                         f"Tool `{tool}` not in list of supported tools or classifications."
                     )
+
         normalized = {'tools': tools, 'classifications': classifications}
+        print(normalized)
         return self.create_ontology(name, normalized)
 
     def create_ontology(self, name, normalized):
@@ -857,8 +859,8 @@ def create_root_schema_node(self, normalized):
         # The OntologyBuilder automatically assigns colors when calling asdict() but Tools and Classifications do not.
         # So we check here to prevent getting 500 erros
 
-        if 'color' not in normalized_json:
+        if 'color' not in normalized:
             raise KeyError("Must provide color.")
-        params = {'data': {'normalized': json.dumps(normalized_json)}}
+        params = {'data': {'normalized': json.dumps(normalized)}}
         res = self.execute(query_str, params)
         return Entity.RootSchemaNode(self, res['upsertRootSchemaNode'])

labelbox/schema/project.py

@@ -429,10 +429,25 @@ def setup_editor(self, ontology):
         Args:
             ontology (Ontology): The ontology to attach to the project
         """
-        labeling_front_end = next(
+        labeling_frontend = next(
             self.client.get_labeling_frontends(
                 where=Entity.LabelingFrontend.name == "Editor"))
-        self.labeling_frontend.connect(labeling_front_end)
+        self.labeling_frontend.connect(labeling_frontend)
+
+        LFO = Entity.LabelingFrontendOptions
+        self.client._create(
+            LFO, {
+                LFO.project:
+                    self,
+                LFO.labeling_frontend:
+                    labeling_frontend,
+                LFO.customization_options:
+                    json.dumps({
+                        "tools": [],
+                        "classifications": []
+                    })
+            })
+
         query_str = """mutation ConnectOntologyPyApi($projectId: ID!, $ontologyId: ID!){
             project(where: {id: $projectId}) {connectOntology(ontologyId: $ontologyId) {id}}}"""
         self.client.execute(query_str, {