Add missing colors (#165)

ufal · Dec 4, 2024 · ed12c0c · ed12c0c
1 parent 527fda1
commit ed12c0c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/factgenie/datasets/propaganda_techniques.py b/factgenie/datasets/propaganda_techniques.py
@@ -39,6 +39,8 @@
     "#8f0e42",  # Dark Pink
     "#095700",  # Dark Green
     "#FFD700",  # Gold
+    "#FF4500",  # Orange Red
+    "#8B4513",  # Saddle Brown
 ]
 
 
@@ -270,6 +272,7 @@ def download(
 
         short_categories = [{"name": c["name"], "color": c["color"], "description": ""} for c in PCT_span_categories]
         categories_names = [c["name"] for c in short_categories]
+
         # save outputs
         outputs_jsonl_parent = out_download_dir / dataset_id
         outputs_jsonl_parent.mkdir(parents=True, exist_ok=True)
@@ -360,7 +363,8 @@ def _load_example_annotations(
                 try:
                     type_idx = categories_names.index(category)
                 except ValueError:
-                    __import__("ipdb").set_trace()
+                    logger.warning(f"Category {category} not found in categories_names")
+                    continue
 
                 annotation_d = {
                     "reason": "",