query for ontology

Author: Matt Sokoloff

labelbox/client.py

@@ -28,6 +28,7 @@
 from labelbox.schema.labeling_frontend import LabelingFrontend
 from labelbox.schema.iam_integration import IAMIntegration
 from labelbox.schema import role
+from labelbox.orm.model import Entity
 
 logger = logging.getLogger(__name__)
 
@@ -709,3 +710,22 @@ def get_data_row_ids_for_external_ids(
                 })['externalIdsToDataRowIds']:
                 result[row['externalId']].append(row['dataRowId'])
         return result
+
+    def get_ontology(self, ontology_id):
+        return self._get_single(Entity.Ontology, ontology_id)
+
+    def get_ontologies(self, name_contains: str):
+        query_str = """query getOntologiesPyApi($search: String, $filter: OntologyFilter, $from : String, $first: PageSize){
+            ontologies(where: {filter: $filter, search: $search}, after: $from, first: $first){
+                nodes {%s}
+                nextCursor
+            }
+        }
+        """ % query.results_query_part(Entity.Ontology)
+        res = PaginatedCollection(
+            self, query_str, {'search' : name_contains, 'filter' :{'status' : 'ALL'}}, ['ontologies', 'nodes'],
+            Entity.Ontology, ['ontologies', 'nextCursor'])
+        return res
+
+
+

labelbox/schema/dataset.py

@@ -360,12 +360,12 @@ def data_row_for_external_id(self, external_id):
         return data_rows[0]
 
     def export_data_rows(self, timeout_seconds=120):
-        """ Returns a generator that produces all data rows that are currently 
+        """ Returns a generator that produces all data rows that are currently
         attached to this dataset.
-            
-        Note: For efficiency, the data are cached for 30 minutes. Newly created data rows will not appear 
+
+        Note: For efficiency, the data are cached for 30 minutes. Newly created data rows will not appear
         until the end of the cache period.
-            
+
         Args:
             timeout_seconds (float): Max waiting time, in seconds.
         Returns:

labelbox/schema/ontology.py

@@ -233,6 +233,11 @@ def add_classification(self, classification: Classification):
         self.classifications.append(classification)
 
 
+class FeatureSchema(DbObject):
+    ...
+
+
+
 class Ontology(DbObject):
     """An ontology specifies which tools and classifications are available
     to a project. This is read only for now.