Skip to content

Commit 60c20f9

Browse files
committed
added delete path since it is small
1 parent 052fe20 commit 60c20f9

File tree

2 files changed

+18
-0
lines changed

2 files changed

+18
-0
lines changed

libs/labelbox/src/labelbox/schema/annotation_import.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -587,6 +587,20 @@ def parent_id(self) -> str:
587587
"""
588588
return self.project().uid
589589

590+
def delete(self) -> None:
591+
"""
592+
Deletes a MALPredictionImport job
593+
"""
594+
595+
query_string = """
596+
mutation deleteModelAssistedLabelingPredictionImportPyApi($id: ID!) {
597+
deleteModelAssistedLabelingPredictionImport(where: { id: $id }) {
598+
id
599+
}
600+
}
601+
"""
602+
self.client.execute(query_string, {"id": self.uid})
603+
590604
@classmethod
591605
def create_from_file(
592606
cls, client: "labelbox.Client", project_id: str, name: str, path: str

libs/labelbox/tests/data/annotation_import/test_mal_prediction_import.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,3 +78,7 @@ def test_get_mal_import_jobs_from_project(client, configured_project):
7878
label_imports = list(configured_project.get_mal_prediction_imports())
7979
assert len(label_imports) == 1
8080
assert label_imports[0].input_file_url == url
81+
82+
label_imports[0].delete()
83+
label_imports = list(configured_project.get_mal_prediction_imports())
84+
assert len(label_imports) == 0

0 commit comments

Comments
 (0)