tests passing

Author: Matt Sokoloff

labelbox/schema/annotation_import.py

@@ -219,7 +219,7 @@ def create_from_url(cls, client: "labelbox.Client", model_run_id: str,
                                    "fileUrl": url,
                                    "modelRunId": model_run_id,
                                    'name': name
-                               })["modelErrorAnalysisPredictionImport"])
+                               })["createModelErrorAnalysisPredictionImport"])
         else:
             raise ValueError(f"Url {url} is not reachable")
 
@@ -252,11 +252,11 @@ def from_name(cls, client: "labelbox.Client", model_run_id: str,
         response = response["modelErrorAnalysisPredictionImport"]
         if as_json:
             return response
-        return cls(client, response["modelErrorAnalysisPredictionImport"])
+        return cls(client, response)
 
     @classmethod
     def _get_url_mutation(cls) -> str:
-        return """mutation createMeaPredictionImportPyApi($modelRunId : ID!, $name: String!, $fileUrl: String!) {
+        return """mutation createMEAPredictionImportPyApi($modelRunId : ID!, $name: String!, $fileUrl: String!) {
             createModelErrorAnalysisPredictionImport(data: {
                 modelRunId: $modelRunId
                 name: $name
@@ -266,7 +266,7 @@ def _get_url_mutation(cls) -> str:
 
     @classmethod
     def _get_file_mutation(cls) -> str:
-        return """mutation createMeaPredictionImportPyApi($modelRunId : ID!, $name: String!, $file: Upload!, $contentLength: Int!) {
+        return """mutation createMEAPredictionImportPyApi($modelRunId : ID!, $name: String!, $file: Upload!, $contentLength: Int!) {
             createModelErrorAnalysisPredictionImport(data: {
                 modelRunId: $modelRunId name: $name filePayload: { file: $file, contentLength: $contentLength}
         }) {%s}
@@ -406,7 +406,7 @@ def from_name(cls, client: "labelbox.Client", project_id: str,
         response = response["modelAssistedLabelingPredictionImport"]
         if as_json:
             return response
-        return cls(client, response["modelAssistedLabelingPredictionImport"])
+        return cls(client, response)
 
     @classmethod
     def _get_url_mutation(cls) -> str:

tests/integration/bulk_import/conftest.py

@@ -1,4 +1,3 @@
-from labelbox.schema.bulk_import_request import BulkImportRequest
 import uuid
 
 import pytest
@@ -305,7 +304,7 @@ def model_run(client, rand_gen, configured_project, annotation_submit_fn,
     configured_project.enable_model_assisted_labeling()
     ontology = configured_project.ontology()
 
-    upload_task = BulkImportRequest.create_from_objects(
+    upload_task = MALPredictionImport.create_from_objects(
         client, configured_project.uid, f'mal-import-{uuid.uuid4()}',
         model_run_predictions)
     upload_task.wait_until_done()