allow that label column is missing

tira-io · Nov 13, 2024 · 7b78da6 · 7b78da6
1 parent 084c9c0
commit 7b78da6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cli/__main__.py b/cli/__main__.py
@@ -479,7 +479,9 @@ def prepare_relevance_judgments(
             )
             # TODO: Instead update existing data and migrate annotations?
 
-        group_pool = pool[pool["group"] == group]
+        group_pool = pool[pool["group"] == group].copy()
+        if 'label' not in group_pool.columns:
+            group_pool['label'] = group_pool['query'].map(lambda i: [])
         echo(
             f"Uploading {len(group_pool)} documents to project '{project.name}'...")
         with NamedTemporaryFile() as tmp_file: