gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-04-03 00:14:33 +08:00
parent 419432d156
commit 84d6dee34d
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/kg/networkx_impl.py
View File

@@ -316,7 +316,9 @@ class NetworkXStorage(BaseGraphStorage):
if depth < max_depth: if depth < max_depth:
# Add neighbor nodes to queue with incremented depth # Add neighbor nodes to queue with incremented depth
neighbors = list(graph.neighbors(current)) neighbors = list(graph.neighbors(current))
queue.extend([(n, depth + 1) for n in neighbors if n not in visited]) queue.extend(
[(n, depth + 1) for n in neighbors if n not in visited]
)
# Check if graph is truncated - if we still have nodes in the queue # Check if graph is truncated - if we still have nodes in the queue
# and we've reached max_nodes, then the graph is truncated # and we've reached max_nodes, then the graph is truncated