Merge pull request #436 from DagsHub/bug/fiftyone-annotations

FIftyOne: Unpack the new tuple that gets returned from the ls converter
DagsHub · Feb 20, 2024 · 61eacf6 · 61eacf6
2 parents 2162a2e + cf9edcb
commit 61eacf6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dagshub/data_engine/annotation/voxel_conversion.py b/dagshub/data_engine/annotation/voxel_conversion.py
@@ -59,6 +59,8 @@ def add_ls_annotations(sample: "fo.Sample", datapoint: "Datapoint", *annotation_
             for res in ann["result"]:
                 try:
                     converted = import_label_studio_annotation(res)
+                    if type(converted) is tuple:
+                        converted = converted[1]
                     annotations.append(converted)
                 except Exception:
                     logger.warning(f"Couldn't convert LS annotation {ann} to voxel annotation")