@@ -496,13 +496,6 @@ def _delete(self, data=None):
496
496
labeledshape_ids = [shape ["id" ] for shape in data ["shapes" ]]
497
497
labeledtrack_ids = [track ["id" ] for track in data ["tracks" ]]
498
498
499
- # It is not important for us that data had some "invalid" objects
500
- # which were skipped (not actually deleted). The main idea is to
501
- # say that all requested objects are absent in DB after the method.
502
- self .ir_data .tags = data ['tags' ]
503
- self .ir_data .shapes = data ['shapes' ]
504
- self .ir_data .tracks = data ['tracks' ]
505
-
506
499
for labeledimage_ids_chunk in take_by (labeledimage_ids , chunk_size = 1000 ):
507
500
self ._delete_job_labeledimages (labeledimage_ids_chunk )
508
501
@@ -513,11 +506,13 @@ def _delete(self, data=None):
513
506
self ._delete_job_labeledtracks (labeledtrack_ids_chunk )
514
507
515
508
deleted_data = {
509
+ "version" : self .ir_data .version ,
516
510
"tags" : data ["tags" ],
517
511
"shapes" : data ["shapes" ],
518
512
"tracks" : data ["tracks" ],
519
513
}
520
514
515
+ self .reset ()
521
516
return deleted_data
522
517
523
518
def delete (self , data = None ):
@@ -526,6 +521,7 @@ def delete(self, data=None):
526
521
self ._set_updated_date ()
527
522
528
523
handle_annotations_change (self .db_job , deleted_data , "delete" )
524
+ return deleted_data
529
525
530
526
@staticmethod
531
527
def _extend_attributes (attributeval_set , default_attribute_values ):
@@ -1033,7 +1029,7 @@ def patch_job_data(pk, data: AnnotationIR | dict, action: PatchAction, *, db_job
1033
1029
elif action == PatchAction .UPDATE :
1034
1030
annotation .update (data )
1035
1031
elif action == PatchAction .DELETE :
1036
- annotation .delete (data )
1032
+ return annotation .delete (data )
1037
1033
1038
1034
return annotation .data
1039
1035
0 commit comments