Formatting

Author: tomislav-peharda

labelbox/schema/model.py

@@ -43,9 +43,5 @@ def delete_model(self):
         """
         ids_param = "ids"
         query_str = """mutation DeleteModelPyApi($%s: ID!) {
-            deleteModels(where: {ids: [$%s]})}""" % (
-            ids_param, ids_param
-        )
-        self.client.execute(query_str, {
-            ids_param: str(self.uid)
-        })
+            deleteModels(where: {ids: [$%s]})}""" % (ids_param, ids_param)
+        self.client.execute(query_str, {ids_param: str(self.uid)})

labelbox/schema/model_run.py

@@ -53,8 +53,8 @@ def add_predictions(
                 return MEAPredictionImport.create_from_file(
                     path=str(predictions), **kwargs)
             else:
-                return MEAPredictionImport.create_from_url(url=str(predictions),
-                                                           **kwargs)
+                return MEAPredictionImport.create_from_url(
+                    url=str(predictions), **kwargs)
         elif isinstance(predictions, Iterable):
             return MEAPredictionImport.create_from_objects(
                 predictions=predictions, **kwargs)
@@ -82,12 +82,8 @@ def delete_model_run(self):
         """
         ids_param = "ids"
         query_str = """mutation DeleteModelRunPyApi($%s: ID!) {
-            deleteModelRuns(where: {ids: [$%s]})}""" % (
-            ids_param, ids_param
-        )
-        self.client.execute(query_str, {
-            ids_param: str(self.uid)
-        })
+            deleteModelRuns(where: {ids: [$%s]})}""" % (ids_param, ids_param)
+        self.client.execute(query_str, {ids_param: str(self.uid)})
 
     def delete_annotation_groups(self, data_row_ids):
         """ Deletes annotation groups by data row ids for a model run.
@@ -101,8 +97,8 @@ def delete_annotation_groups(self, data_row_ids):
         data_row_ids_param = "dataRowIds"
         query_str = """mutation DeleteModelRunDataRowsPyApi($%s: ID!, $%s: [ID!]!) {
             deleteModelRunDataRows(where: {modelRunId: $%s, dataRowIds: $%s})}""" % (
-            model_run_id_param, data_row_ids_param, model_run_id_param, data_row_ids_param
-        )
+            model_run_id_param, data_row_ids_param, model_run_id_param,
+            data_row_ids_param)
         self.client.execute(query_str, {
             model_run_id_param: self.uid,
             data_row_ids_param: data_row_ids
@@ -123,4 +119,3 @@ def url(self):
         app_url = self.client.app_url
         endpoint = f"{app_url}/models/{self.model_id}/{self.model_run_id}/AllDatarowsSlice/{self.uid}?view=carousel"
         return endpoint
-

tests/integration/mal_and_mea/test_model_run.py

@@ -39,18 +39,19 @@ def test_model_run_delete(client, model_run):
     assert len(before) == len(after) + 1
 
 
-def test_model_run_annotation_groups_delete(client, model_run_annotation_groups):
+def test_model_run_annotation_groups_delete(client,
+                                            model_run_annotation_groups):
     models = list(client.get_models())
     model = models[0]
     model_runs = list(model.model_runs())
     model_run = model_runs[0]
 
     before = list(model_run.annotation_groups())
     annotation_group = before[0]
-    
+
     data_row_id = annotation_group.data_row().uid
     model_run.delete_annotation_groups(data_row_ids=[data_row_id])
 
     after = list(model_run.annotation_groups())
-    
+
     assert len(before) == len(after) + 1