format

Author: Matt Sokoloff

labelbox/client.py

@@ -753,7 +753,11 @@ def get_feature_schema(self, feature_schema_id):
         query_str = """query rootSchemaNodePyApi($rootSchemaNodeWhere: RootSchemaNodeWhere!){
               rootSchemaNode(where: $rootSchemaNodeWhere){%s}
         }""" % query.results_query_part(Entity.FeatureSchema)
-        res = self.execute(query_str, {'rootSchemaNodeWhere': {'featureSchemaId' : feature_schema_id}})['rootSchemaNode']
+        res = self.execute(
+            query_str,
+            {'rootSchemaNodeWhere': {
+                'featureSchemaId': feature_schema_id
+            }})['rootSchemaNode']
         res['id'] = res['normalized']['featureSchemaId']
         return Entity.FeatureSchema(self, res)
 
@@ -774,6 +778,7 @@ def get_feature_schemas(self, name_contains):
         }
         """ % query.results_query_part(Entity.FeatureSchema)
         params = {'search': name_contains, 'filter': {'status': 'ALL'}}
+
         def rootSchemaPayloadToFeatureSchema(client, payload):
             # Technically we are querying for a Schema Node.
             # But the features are the same so we just grab the feature schema id
@@ -785,8 +790,7 @@ def rootSchemaPayloadToFeatureSchema(client, payload):
                                    rootSchemaPayloadToFeatureSchema,
                                    ['rootSchemaNodes', 'nextCursor'])
 
-    def create_ontology_from_feature_schemas(self, name,
-                                               feature_schema_ids):
+    def create_ontology_from_feature_schemas(self, name, feature_schema_ids):
         """
         Creates an ontology from a list of feature schema ids
 
@@ -889,4 +893,3 @@ def create_feature_schema(self, normalized):
         # But the features are the same so we just grab the feature schema id
         res['id'] = res['normalized']['featureSchemaId']
         return Entity.FeatureSchema(self, res)
-

labelbox/schema/ontology.py

@@ -136,7 +136,6 @@ def from_dict(cls, dictionary: Dict[str, Any]):
                    schema_id=dictionary.get("schemaNodeId", None),
                    feature_schema_id=dictionary.get("featureSchemaId", None))
 
-
     def asdict(self) -> Dict[str, Any]:
         if self.class_type in self._REQUIRES_OPTIONS \
                 and len(self.options) < 1:

tests/integration/test_ontology.py

@@ -271,9 +271,8 @@ def test_feature_schema_create_read(client, rand_gen):
 
     time.sleep(3)  # Slight delay for searching
     queried_feature_schemas = list(client.get_feature_schemas(name))
-    assert [
-        feature_schema.name for feature_schema in queried_feature_schemas
-    ] == [name]
+    assert [feature_schema.name for feature_schema in queried_feature_schemas
+           ] == [name]
     queried_feature_schema = queried_feature_schemas[0]
 
     for attr in Entity.FeatureSchema.fields():
@@ -291,8 +290,7 @@ def test_ontology_create_read(client, rand_gen):
         'color': 'black',
         'classifications': [],
     }
-    feature_schema = client.create_feature_schema(
-        feature_schema_cat_normalized)
+    feature_schema = client.create_feature_schema(feature_schema_cat_normalized)
     created_ontology = client.create_ontology_from_feature_schemas(
         name=ontology_name, feature_schema_ids=[feature_schema.uid])
     tool_normalized = created_ontology.normalized['tools'][0]