Kevin Hu commited on
Commit
5b47d62
·
1 Parent(s): c786948

fix empty children in mindmap (#2418)

Browse files

### What problem does this PR solve?

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Files changed (1) hide show
  1. graphrag/mind_map_extractor.py +1 -1
graphrag/mind_map_extractor.py CHANGED
@@ -66,7 +66,7 @@ class MindMapExtractor:
66
  obj = [obj]
67
  if isinstance(obj, list):
68
  for i in obj: keyset.add(i)
69
- return [{"id": re.sub(r"\*+", "", i), "children": []} for i in obj]
70
  arr = []
71
  for k, v in obj.items():
72
  k = self._key(k)
 
66
  obj = [obj]
67
  if isinstance(obj, list):
68
  for i in obj: keyset.add(i)
69
+ return [{"id": re.sub(r"\*+", "", i), "children": []} for i in obj if re.sub(r"\*+", "", i)]
70
  arr = []
71
  for k, v in obj.items():
72
  k = self._key(k)