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-05-11 11:24:52 +08:00
parent 68653f853a
commit 9ec9579a95
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lightrag/operate.py
View File

@@ -1610,9 +1610,11 @@ async def _find_most_related_edges_from_entities(
edge_props = edge_data_dict.get(pair) edge_props = edge_data_dict.get(pair)
if edge_props is not None: if edge_props is not None:
if "weight" not in edge_props: if "weight" not in edge_props:
logger.warning(f"Edge {pair} missing 'weight' attribute, using default value 0.0") logger.warning(
f"Edge {pair} missing 'weight' attribute, using default value 0.0"
)
edge_props["weight"] = 0.0 edge_props["weight"] = 0.0
combined = { combined = {
"src_tgt": pair, "src_tgt": pair,
"rank": edge_degrees_dict.get(pair, 0), "rank": edge_degrees_dict.get(pair, 0),
@@ -1675,9 +1677,11 @@ async def _get_edge_data(
edge_props = edge_data_dict.get(pair) edge_props = edge_data_dict.get(pair)
if edge_props is not None: if edge_props is not None:
if "weight" not in edge_props: if "weight" not in edge_props:
logger.warning(f"Edge {pair} missing 'weight' attribute, using default value 0.0") logger.warning(
f"Edge {pair} missing 'weight' attribute, using default value 0.0"
)
edge_props["weight"] = 0.0 edge_props["weight"] = 0.0
# Use edge degree from the batch as rank. # Use edge degree from the batch as rank.
combined = { combined = {
"src_id": k["src_id"], "src_id": k["src_id"],