diff --git a/label_studio_converter/converter.py b/label_studio_converter/converter.py index 86bac55..fe55441 100644 --- a/label_studio_converter/converter.py +++ b/label_studio_converter/converter.py @@ -554,7 +554,7 @@ def add_image(images, width, height, image_id, image_path): for label in labels: category_name = None - for key in ['rectanglelabels', 'polygonlabels', 'brushlabels' 'labels']: + for key in ['rectanglelabels', 'polygonlabels', 'brushlabels', 'labels']: if key in label and len(label[key]) > 0: category_name = label[key][0] break