add deletion of MEA predictions to stop sentry errors

Author: jtsodapop

tests/integration/annotation_import/test_mea_prediction_import.py

@@ -10,7 +10,7 @@
 """
 
 
-def test_create_from_url(model_run_with_model_run_data_rows,
+def test_create_from_url(client, model_run_with_model_run_data_rows,
                          annotation_import_test_helpers):
     name = str(uuid.uuid4())
     url = "https://storage.googleapis.com/labelbox-public-bucket/predictions_test_v2.ndjson"
@@ -19,9 +19,20 @@ def test_create_from_url(model_run_with_model_run_data_rows,
     assert annotation_import.model_run_id == model_run_with_model_run_data_rows.uid
     annotation_import_test_helpers.check_running_state(annotation_import, name,
                                                        url)
-
-
-def test_create_from_objects(model_run_with_model_run_data_rows,
+    import_id = client.execute(
+        """query get_mea_importPyApi($model_run_id: ID!) {
+        modelErrorAnalysisPredictionImports(where: {modelRunId: $model_run_id}) {
+            id
+            }}""", {"model_run_id": annotation_import.parent_id
+                    })['modelErrorAnalysisPredictionImports'][0]['id']
+    client.execute(
+        """mutation deleteModelErrorAnalysisPredictionImportPyApi($import_id: ID!) {
+        deleteModelErrorAnalysisPredictionImport(where: {id: $import_id}) {
+            id
+        }}""", {"import_id": import_id})
+
+
+def test_create_from_objects(client, model_run_with_model_run_data_rows,
                              object_predictions,
                              annotation_import_test_helpers):
     name = str(uuid.uuid4())
@@ -33,9 +44,21 @@ def test_create_from_objects(model_run_with_model_run_data_rows,
     annotation_import_test_helpers.check_running_state(annotation_import, name)
     annotation_import_test_helpers.assert_file_content(
         annotation_import.input_file_url, object_predictions)
-
-
-def test_create_from_local_file(tmp_path, model_run_with_model_run_data_rows,
+    import_id = client.execute(
+        """query get_mea_importPyApi($model_run_id: ID!) {
+        modelErrorAnalysisPredictionImports(where: {modelRunId: $model_run_id}) {
+            id
+            }}""", {"model_run_id": annotation_import.parent_id
+                    })['modelErrorAnalysisPredictionImports'][0]['id']
+    client.execute(
+        """mutation deleteModelErrorAnalysisPredictionImportPyApi($import_id: ID!) {
+        deleteModelErrorAnalysisPredictionImport(where: {id: $import_id}) {
+            id
+        }}""", {"import_id": import_id})
+
+
+def test_create_from_local_file(client, tmp_path,
+                                model_run_with_model_run_data_rows,
                                 object_predictions,
                                 annotation_import_test_helpers):
     name = str(uuid.uuid4())
@@ -51,6 +74,17 @@ def test_create_from_local_file(tmp_path, model_run_with_model_run_data_rows,
     annotation_import_test_helpers.check_running_state(annotation_import, name)
     annotation_import_test_helpers.assert_file_content(
         annotation_import.input_file_url, object_predictions)
+    import_id = client.execute(
+        """query get_mea_importPyApi($model_run_id: ID!) {
+        modelErrorAnalysisPredictionImports(where: {modelRunId: $model_run_id}) {
+            id
+            }}""", {"model_run_id": annotation_import.parent_id
+                    })['modelErrorAnalysisPredictionImports'][0]['id']
+    client.execute(
+        """mutation deleteModelErrorAnalysisPredictionImportPyApi($import_id: ID!) {
+        deleteModelErrorAnalysisPredictionImport(where: {id: $import_id}) {
+            id
+        }}""", {"import_id": import_id})
 
 
 def test_get(client, model_run_with_model_run_data_rows,
@@ -66,6 +100,17 @@ def test_get(client, model_run_with_model_run_data_rows,
     assert annotation_import.model_run_id == model_run_with_model_run_data_rows.uid
     annotation_import_test_helpers.check_running_state(annotation_import, name,
                                                        url)
+    import_id = client.execute(
+        """query get_mea_importPyApi($model_run_id: ID!) {
+        modelErrorAnalysisPredictionImports(where: {modelRunId: $model_run_id}) {
+            id
+            }}""", {"model_run_id": annotation_import.parent_id
+                    })['modelErrorAnalysisPredictionImports'][0]['id']
+    client.execute(
+        """mutation deleteModelErrorAnalysisPredictionImportPyApi($import_id: ID!) {
+        deleteModelErrorAnalysisPredictionImport(where: {id: $import_id}) {
+            id
+        }}""", {"import_id": import_id})
 
 
 @pytest.mark.slow