diff --git a/graphiti_core/utils/bulk_utils.py b/graphiti_core/utils/bulk_utils.py index d1167dba..321cbd33 100644 --- a/graphiti_core/utils/bulk_utils.py +++ b/graphiti_core/utils/bulk_utils.py @@ -433,7 +433,7 @@ async def dedupe_edges_bulk( dedupe_tuples: list[tuple[EpisodicNode, EntityEdge, list[EntityEdge]]] = [] for i, edges_i in enumerate(extracted_edges): existing_edges: list[EntityEdge] = [] - for j, edges_j in enumerate(extracted_edges): + for edges_j in extracted_edges: existing_edges += edges_j for edge in edges_i: diff --git a/tests/utils/maintenance/test_bulk_utils.py b/tests/utils/maintenance/test_bulk_utils.py index 4b6b093a..2616c79a 100644 --- a/tests/utils/maintenance/test_bulk_utils.py +++ b/tests/utils/maintenance/test_bulk_utils.py @@ -313,7 +313,7 @@ async def test_dedupe_edges_bulk_deduplicates_within_episode(monkeypatch): episodes=[episode.uuid], ) - edges_by_episode = await bulk_utils.dedupe_edges_bulk( + await bulk_utils.dedupe_edges_bulk( clients, [[edge1, edge2, edge3]], [(episode, [])], @@ -325,6 +325,6 @@ async def test_dedupe_edges_bulk_deduplicates_within_episode(monkeypatch): # Verify that edges were compared against each other (within same episode) # Each edge should have been compared against all 3 edges (including itself, which gets filtered) assert len(comparisons_made) == 3 - for edge_uuid, compared_against in comparisons_made: + for _, compared_against in comparisons_made: # Each edge should have access to all 3 edges as candidates assert len(compared_against) >= 2 # At least 2 others (self is filtered out)