File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed
tests/integration/annotation_import Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -200,8 +200,7 @@ def export_labels(
200
200
None is returned.
201
201
"""
202
202
sleep_time = 2
203
- query_str = """
204
- mutation exportModelRunAnnotationsPyApi($modelRunId: ID!) {
203
+ query_str = """mutation exportModelRunAnnotationsPyApi($modelRunId: ID!) {
205
204
exportModelRunAnnotations(data: {modelRunId: $modelRunId}) {
206
205
downloadUrl createdAt status
207
206
}
Original file line number Diff line number Diff line change @@ -73,3 +73,8 @@ def test_model_run_upsert_data_rows_with_existing_labels(
73
73
])
74
74
assert n_data_rows == len (
75
75
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
You can’t perform that action at this time.
0 commit comments