handle no-label classification case

jainanisha90 · Mar 26, 2024 · 74e214c · 74e214c
1 parent b63cdbf
commit 74e214c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/fiftyone/utils/hf_hub.py b/fiftyone/utils/hf_hub.py
@@ -806,7 +806,9 @@ def _build_label_field_converter(
     def convert_classification_field(sample_dict, row):
         row_content = row["row"]
         label_index = row_content[field_name]
-        label = (feature["names"][label_index],)
+        if label_index == -1:
+            return
+        label = feature["names"][label_index]
         if isinstance(label, tuple):
             label = label[0]
         sample_dict[field_name] = fol.Classification(label=str(label))