@@ -168,10 +168,12 @@ def test_dataset_append_async(dataset: Dataset):
168
168
job = dataset .append (make_dataset_items (), asynchronous = True )
169
169
job .sleep_until_complete ()
170
170
status = job .status ()
171
+ status ["message" ]["Payload" ] = ""
171
172
assert status == {
172
173
"job_id" : job .id ,
173
174
"status" : "Completed" ,
174
175
"message" : {
176
+ "Payload" : "" ,
175
177
"image_upload_step" : {"errored" : 0 , "pending" : 0 , "completed" : 5 },
176
178
"started_image_processing" : f"Dataset: { dataset .id } , Job: { job .id } " ,
177
179
"ingest_to_reupload_queue" : {
@@ -200,10 +202,13 @@ def test_dataset_append_async_with_1_bad_url(dataset: Dataset):
200
202
job = dataset .append (ds_items , asynchronous = True )
201
203
with pytest .raises (JobError ):
202
204
job .sleep_until_complete ()
203
- assert job .status () == {
205
+ status = job .status ()
206
+ status ["message" ]["Payload" ] = ""
207
+ assert status == {
204
208
"job_id" : f"{ job .id } " ,
205
209
"status" : "Errored" ,
206
210
"message" : {
211
+ "Payload" : "" ,
207
212
"final_error" : (
208
213
"One or more of the images you attempted to upload did not process"
209
214
" correctly. Please see the status for an overview and the errors for "
@@ -273,7 +278,7 @@ def test_dataset_export_autotag_scores(CLIENT):
273
278
def test_annotate_async (dataset : Dataset ):
274
279
dataset .append (make_dataset_items ())
275
280
semseg = SegmentationAnnotation .from_json (TEST_SEGMENTATION_ANNOTATIONS [0 ])
276
- polygon = PolygonAnnotation ( ** TEST_POLYGON_ANNOTATIONS [0 ])
281
+ polygon = PolygonAnnotation . from_json ( TEST_POLYGON_ANNOTATIONS [0 ])
277
282
bbox = BoxAnnotation (** TEST_BOX_ANNOTATIONS [0 ])
278
283
279
284
job : AsyncJob = dataset .annotate (
@@ -306,7 +311,7 @@ def test_annotate_async(dataset: Dataset):
306
311
def test_annotate_async_with_error (dataset : Dataset ):
307
312
dataset .append (make_dataset_items ())
308
313
semseg = SegmentationAnnotation .from_json (TEST_SEGMENTATION_ANNOTATIONS [0 ])
309
- polygon = PolygonAnnotation ( ** TEST_POLYGON_ANNOTATIONS [0 ])
314
+ polygon = PolygonAnnotation . from_json ( TEST_POLYGON_ANNOTATIONS [0 ])
310
315
bbox = BoxAnnotation (** TEST_BOX_ANNOTATIONS [0 ])
311
316
bbox .reference_id = "fake_garbage"
312
317
0 commit comments