@@ -168,10 +168,12 @@ def test_dataset_append_async(dataset: Dataset):
168168 job = dataset .append (make_dataset_items (), asynchronous = True )
169169 job .sleep_until_complete ()
170170 status = job .status ()
171+ status ["message" ]["Payload" ] = ""
171172 assert status == {
172173 "job_id" : job .id ,
173174 "status" : "Completed" ,
174175 "message" : {
176+ "Payload" : "" ,
175177 "image_upload_step" : {"errored" : 0 , "pending" : 0 , "completed" : 5 },
176178 "started_image_processing" : f"Dataset: { dataset .id } , Job: { job .id } " ,
177179 "ingest_to_reupload_queue" : {
@@ -200,10 +202,13 @@ def test_dataset_append_async_with_1_bad_url(dataset: Dataset):
200202 job = dataset .append (ds_items , asynchronous = True )
201203 with pytest .raises (JobError ):
202204 job .sleep_until_complete ()
203- assert job .status () == {
205+ status = job .status ()
206+ status ["message" ]["Payload" ] = ""
207+ assert status == {
204208 "job_id" : f"{ job .id } " ,
205209 "status" : "Errored" ,
206210 "message" : {
211+ "Payload" : "" ,
207212 "final_error" : (
208213 "One or more of the images you attempted to upload did not process"
209214 " correctly. Please see the status for an overview and the errors for "
@@ -273,7 +278,7 @@ def test_dataset_export_autotag_scores(CLIENT):
273278def test_annotate_async (dataset : Dataset ):
274279 dataset .append (make_dataset_items ())
275280 semseg = SegmentationAnnotation .from_json (TEST_SEGMENTATION_ANNOTATIONS [0 ])
276- polygon = PolygonAnnotation ( ** TEST_POLYGON_ANNOTATIONS [0 ])
281+ polygon = PolygonAnnotation . from_json ( TEST_POLYGON_ANNOTATIONS [0 ])
277282 bbox = BoxAnnotation (** TEST_BOX_ANNOTATIONS [0 ])
278283
279284 job : AsyncJob = dataset .annotate (
@@ -306,7 +311,7 @@ def test_annotate_async(dataset: Dataset):
306311def test_annotate_async_with_error (dataset : Dataset ):
307312 dataset .append (make_dataset_items ())
308313 semseg = SegmentationAnnotation .from_json (TEST_SEGMENTATION_ANNOTATIONS [0 ])
309- polygon = PolygonAnnotation ( ** TEST_POLYGON_ANNOTATIONS [0 ])
314+ polygon = PolygonAnnotation . from_json ( TEST_POLYGON_ANNOTATIONS [0 ])
310315 bbox = BoxAnnotation (** TEST_BOX_ANNOTATIONS [0 ])
311316 bbox .reference_id = "fake_garbage"
312317
0 commit comments