dataloop comment bugfixed

Author: agyurjin

superannotate/input_converters/converters/coco_converters/coco_strategies.py

@@ -239,13 +239,6 @@ def sa_to_output_format(self):
         write_to_json(
             self.output_dir / '{}.json'.format(self.dataset_name), out_json
         )
-        # json_data = json.dumps(out_json, indent=4)
-        # with open(
-        #     self.output_dir / '{}.json'.format(self.dataset_name), 'w+'
-        # ) as coco_json:
-
-        #     coco_json.write(json_data)
-
         self.set_num_converted(len(jsons))
 
 
@@ -370,11 +363,4 @@ def sa_to_output_format(self):
         write_to_json(
             self.output_dir / '{}.json'.format(self.dataset_name), out_json
         )
-        # json_data = json.dumps(out_json, indent=4)
-
-        # with open(
-        #     self.output_dir / '{}.json'.format(self.dataset_name), 'w+'
-        # ) as coco_json:
-        #     coco_json.write(json_data)
-
         self.set_num_converted(len(out_json['images']))

superannotate/input_converters/converters/dataloop_converters/dataloop_to_sa_vector.py

@@ -76,18 +76,18 @@ def dataloop_to_sa(input_dir, task, output_dir):
                 )
                 sa_instances.append(sa_obj)
             elif ann['type'] == comment_type:
+                points = (
+                    ann['coordinates']['box'][0]['x'],
+                    ann['coordinates']['box'][0]['y']
+                )
                 comments = []
                 for note in ann['coordinates']['note']['messages']:
                     comments.append(
                         {
                             'text': note['body'],
-                            'id': note['creator']
+                            'email': note['creator']
                         }
                     )
-                    points = (
-                        ann['coordinates']['box'][0]['x'],
-                        ann['coordinates']['box'][0]['y']
-                    )
                     sa_comment = _create_comment(points, comments)
                 sa_comments.append(sa_comment)
             elif ann['type'] == tags_type:

superannotate/input_converters/converters/sa_json_helper.py

@@ -79,7 +79,7 @@ def _create_comment(points, comments):
         'type': 'comment',
         'x': points[0],
         'y': points[1],
-        'comments': comments
+        'correspondence': comments
     }
     return sa_comment