Skip to content

Commit 9c20ea7

Browse files
committed
add test case
1 parent 7bba5ab commit 9c20ea7

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

labelbox/schema/model_run.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -200,8 +200,7 @@ def export_labels(
200200
None is returned.
201201
"""
202202
sleep_time = 2
203-
query_str = """
204-
mutation exportModelRunAnnotationsPyApi($modelRunId: ID!) {
203+
query_str = """mutation exportModelRunAnnotationsPyApi($modelRunId: ID!) {
205204
exportModelRunAnnotations(data: {modelRunId: $modelRunId}) {
206205
downloadUrl createdAt status
207206
}

tests/integration/annotation_import/test_model_run.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,3 +73,8 @@ def test_model_run_upsert_data_rows_with_existing_labels(
7373
])
7474
assert n_data_rows == len(
7575
list(model_run_with_model_run_data_rows.model_run_data_rows()))
76+
77+
78+
def test_model_run_export_labels(model_run_with_model_run_data_rows):
79+
labels = model_run_with_model_run_data_rows.export_labels(download=True)
80+
assert len(labels) == 3

0 commit comments

Comments
 (0)