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-03-05 12:32:23 +08:00
parent 5e40e4107d
commit 5f12aa3ee4
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lightrag/kg/networkx_impl.py
View File

@@ -299,7 +299,9 @@ class NetworkXStorage(BaseGraphStorage):
start_nodes = set(nodes_to_explore) start_nodes = set(nodes_to_explore)
# Get nodes directly connected to all start nodes # Get nodes directly connected to all start nodes
for start_node in start_nodes: for start_node in start_nodes:
direct_connected_nodes.update(combined_subgraph.neighbors(start_node)) direct_connected_nodes.update(
combined_subgraph.neighbors(start_node)
)
# Remove start nodes from directly connected nodes (avoid duplicates) # Remove start nodes from directly connected nodes (avoid duplicates)
direct_connected_nodes -= start_nodes direct_connected_nodes -= start_nodes
@@ -309,7 +311,9 @@ class NetworkXStorage(BaseGraphStorage):
nodes_to_keep = [ nodes_to_keep = [
node node
for node, degree in combined_subgraph.degree() for node, degree in combined_subgraph.degree()
if node in start_nodes or node in direct_connected_nodes or degree >= min_degree if node in start_nodes
or node in direct_connected_nodes
or degree >= min_degree
] ]
combined_subgraph = combined_subgraph.subgraph(nodes_to_keep) combined_subgraph = combined_subgraph.subgraph(nodes_to_keep)