gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

nit: update variable names

Browse source
This commit is contained in:
lxobr 2025-12-19 15:35:34 +01:00
parent a85df53c74
commit 9808077b4c
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cognee/modules/graph/cognee_graph/CogneeGraph.py
View file

@ -250,12 +250,12 @@ class CogneeGraph(CogneeAbstractGraph):
if not scored_results: if not scored_results:
continue continue
per_query_lists = self._normalize_query_distance_lists( per_query_scored_results = self._normalize_query_distance_lists(
scored_results, query_list_length, f"Collection '{collection_name}'" scored_results, query_list_length, f"Collection '{collection_name}'"
) )
for query_index, scored_list in enumerate(per_query_lists): for query_index, scored_results in enumerate(per_query_scored_results):
for result in scored_list: for result in scored_results:
node_id = str(getattr(result, "id", None)) node_id = str(getattr(result, "id", None))
if not node_id: if not node_id:
continue continue
@ -283,12 +283,12 @@ class CogneeGraph(CogneeAbstractGraph):
if not edge_distances: if not edge_distances:
return None return None
per_query_edge_lists = self._normalize_query_distance_lists( per_query_scored_results = self._normalize_query_distance_lists(
edge_distances, query_list_length, "edge_distances" edge_distances, query_list_length, "edge_distances"
) )
for query_index, scored_list in enumerate(per_query_edge_lists): for query_index, scored_results in enumerate(per_query_scored_results):
for result in scored_list: for result in scored_results:
payload = getattr(result, "payload", None) payload = getattr(result, "payload", None)
if not isinstance(payload, dict): if not isinstance(payload, dict):
continue continue