Spaces:

argilla
/

synthetic-data-generator

Running

sdiazlor commited on Feb 8

Commit

e3b32d4

1 Parent(s): f8f9187

fix: tackle None labels in multilabel

Files changed (1) hide show

src/synthetic_dataset_generator/apps/textcat.py CHANGED Viewed

@@ -195,7 +195,7 @@ def generate_dataset(
                     set(
                         label.lower().strip()
                         for label in x
-                        if label.lower().strip() in labels
                     )
                 )
             else:

                     set(
                         label.lower().strip()
                         for label in x
+                        if isinstance(label, str) and label.lower().strip() in labels
                     )
                 )
             else: