YAPF

Author: Val Brodsky

labelbox/schema/model_run.py

@@ -48,9 +48,9 @@ class Status(Enum):
         FAILED = "FAILED"
 
     def upsert_labels(self,
-        label_ids: Optional[List[str]] = None,
-        project_id: Optional[str] = None, 
-        timeout_seconds=3600):
+                      label_ids: Optional[List[str]] = None,
+                      project_id: Optional[str] = None,
+                      timeout_seconds=3600):
         """ Adds data rows and labels to a Model Run
         Args:
             label_ids (list): label ids to insert
@@ -64,21 +64,20 @@ def upsert_labels(self,
         use_project_id = project_id is not None
 
         if not use_label_ids and not use_project_id:
-            raise ValueError("Must provide at least one label id or a project id")
+            raise ValueError(
+                "Must provide at least one label id or a project id")
 
         if use_label_ids and use_project_id:
             raise ValueError("Must only one of label ids, project id")
 
         if use_label_ids:
             return self._upsert_labels_by_label_ids(label_ids)
-        else: # use_project_id
+        else:  # use_project_id
             return self._upsert_labels_by_project_id(project_id)
 
-
-
     def _upsert_labels_by_label_ids(self,
-        label_ids: List[str],
-        timeout_seconds=3600):
+                                    label_ids: List[str],
+                                    timeout_seconds=3600):
         mutation_name = 'createMEAModelRunLabelRegistrationTask'
         create_task_query_str = """mutation createMEAModelRunLabelRegistrationTaskPyApi($modelRunId: ID!, $labelIds : [ID!]!) {
         %s(where : { id : $modelRunId}, data : {labelIds: $labelIds})}
@@ -98,11 +97,11 @@ def _upsert_labels_by_label_ids(self,
             status_query_str, {'where': {
                 'id': task_id
             }})['MEALabelRegistrationTaskStatus'],
-                                        timeout_seconds=timeout_seconds)
+                                     timeout_seconds=timeout_seconds)
 
     def _upsert_labels_by_project_id(self,
-        project_id: str,
-        timeout_seconds=3600):
+                                     project_id: str,
+                                     timeout_seconds=3600):
         mutation_name = 'createMEAModelRunProjectLabelRegistrationTask'
         create_task_query_str = """mutation createMEAModelRunProjectLabelRegistrationTaskPyApi($modelRunId: ID!, $projectId : ID!) {
         %s(where : { modelRunId : $modelRunId, projectId: $projectId})}
@@ -122,8 +121,7 @@ def _upsert_labels_by_project_id(self,
             status_query_str, {'where': {
                 'id': task_id
             }})['MEALabelRegistrationTaskStatus'],
-                                        timeout_seconds=timeout_seconds)
-
+                                     timeout_seconds=timeout_seconds)
 
     def upsert_data_rows(self,
                          data_row_ids=None,

tests/integration/annotation_import/conftest.py

@@ -439,9 +439,11 @@ def model_run_with_model_run_data_rows(client, configured_project,
     model_run.delete()
     # TODO: Delete resources when that is possible ..
 
+
 @pytest.fixture
 def model_run_with_model_run_all_project_data_rows(client, configured_project,
-                                       model_run_predictions, model_run):
+                                                   model_run_predictions,
+                                                   model_run):
     configured_project.enable_model_assisted_labeling()
 
     upload_task = LabelImport.create_from_objects(
@@ -454,6 +456,7 @@ def model_run_with_model_run_all_project_data_rows(client, configured_project,
     model_run.delete()
     # TODO: Delete resources when that is possible ..
 
+
 class AnnotationImportTestHelpers:
 
     @classmethod

tests/integration/annotation_import/test_mea_prediction_import.py

@@ -37,9 +37,10 @@ def test_create_from_objects(model_run_with_model_run_data_rows,
         annotation_import.input_file_url, object_predictions)
     annotation_import.wait_until_done()
 
-def test_create_from_objects_all_project_labels(model_run_with_model_run_all_project_data_rows,
-                             object_predictions,
-                             annotation_import_test_helpers):
+
+def test_create_from_objects_all_project_labels(
+        model_run_with_model_run_all_project_data_rows, object_predictions,
+        annotation_import_test_helpers):
     name = str(uuid.uuid4())
 
     annotation_import = model_run_with_model_run_all_project_data_rows.add_predictions(
@@ -51,6 +52,7 @@ def test_create_from_objects_all_project_labels(model_run_with_model_run_all_pro
         annotation_import.input_file_url, object_predictions)
     annotation_import.wait_until_done()
 
+
 def test_create_from_label_objects(model_run_with_model_run_data_rows,
                                    object_predictions,
                                    annotation_import_test_helpers):