From 1a41b92f77f4515eeeb34c6ecd541d6dba0bf226 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Tue, 25 Feb 2025 12:58:10 +0800 Subject: [PATCH] More robust community report. (#5328) ### What problem does this PR solve? #5289 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- graphrag/general/community_reports_extractor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/graphrag/general/community_reports_extractor.py b/graphrag/general/community_reports_extractor.py index 7157f0b31..07750e802 100644 --- a/graphrag/general/community_reports_extractor.py +++ b/graphrag/general/community_reports_extractor.py @@ -68,7 +68,7 @@ class CommunityReportsExtractor(Extractor): weight = ents["weight"] ents = ents["nodes"] ent_df = pd.DataFrame(self._get_entity_(ents)).dropna()#[{"entity": n, **graph.nodes[n]} for n in ents]) - if ent_df.empty: + if ent_df.empty or "entity_name" not in ent_df.columns: continue ent_df["entity"] = ent_df["entity_name"] del ent_df["entity_name"]