Add tests

Author: Val Brodsky

labelbox/schema/model_run.py

@@ -77,7 +77,8 @@ def upsert_labels(self,
 
 
     def _upsert_labels_by_label_ids(self,
-        label_ids: List[str]):
+        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})}
@@ -100,10 +101,11 @@ def _upsert_labels_by_label_ids(self,
                                         timeout_seconds=timeout_seconds)
 
     def _upsert_labels_by_project_id(self,
-        project_id: str):
+        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}}
+        %s(where : { modelRunId : $modelRunId, projectId: $projectId})}
         """ % (mutation_name)
 
         res = self.client.execute(create_task_query_str, {

tests/integration/annotation_import/conftest.py

@@ -439,6 +439,20 @@ 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):
+    configured_project.enable_model_assisted_labeling()
+
+    upload_task = LabelImport.create_from_objects(
+        client, configured_project.uid, f"label-import-{uuid.uuid4()}",
+        model_run_predictions)
+    upload_task.wait_until_done()
+    model_run.upsert_labels(project_id=configured_project.uid)
+    time.sleep(3)
+    yield model_run
+    model_run.delete()
+    # TODO: Delete resources when that is possible ..
 
 class AnnotationImportTestHelpers:
 

tests/integration/annotation_import/test_mea_prediction_import.py

@@ -37,6 +37,19 @@ 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):
+    name = str(uuid.uuid4())
+
+    annotation_import = model_run_with_model_run_all_project_data_rows.add_predictions(
+        name=name, predictions=object_predictions)
+
+    assert annotation_import.model_run_id == model_run_with_model_run_all_project_data_rows.uid
+    annotation_import_test_helpers.check_running_state(annotation_import, name)
+    annotation_import_test_helpers.assert_file_content(
+        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,