image_cat_wall: no top cat? skip it

digitalmethodsinitiative · Jan 8, 2025 · 9465cc2 · 9465cc2
1 parent b0889ae
commit 9465cc2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/processors/visualisation/image_category_wall.py b/processors/visualisation/image_category_wall.py
@@ -231,6 +231,9 @@ def process(self):
 							category_type = float
 						# Special case
 						top_cats = post.get("top_categories")
+						if not top_cats:
+							self.dataset.log(f"Post {post.get('id')} has no top categories; skipping")
+							continue
 						top_cat = top_cats.split(",")[0].split(":")[0].strip()
 						top_cat_score = float(top_cats.split(",")[0].split(":")[1].strip().replace("%", ""))
 						post_result = {"id": post.get("id"), "value": top_cat_score, "top_cats": top_cats}