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)
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)
|