From 6ab3d7de5768e6d0ebd039793cef2a2f207b18d7 Mon Sep 17 00:00:00 2001 From: Ryan Michael Date: Thu, 15 Aug 2024 16:58:35 -0400 Subject: [PATCH] lint --- .../ragstack_knowledge_store/graph_store.py | 4 +++- .../tests/integration_tests/test_graph_store.py | 8 ++++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/libs/knowledge-store/ragstack_knowledge_store/graph_store.py b/libs/knowledge-store/ragstack_knowledge_store/graph_store.py index 068d7ee0f..047984567 100644 --- a/libs/knowledge-store/ragstack_knowledge_store/graph_store.py +++ b/libs/knowledge-store/ragstack_knowledge_store/graph_store.py @@ -484,7 +484,9 @@ def fetch_initial_candidates() -> None: if tag_filter.len() == 0: outgoing_tags[row.content_id] = set(row.link_to_tags or []) else: - outgoing_tags[row.content_id] = tag_filter.intersection(set(row.link_to_tags or [])) + outgoing_tags[row.content_id] = tag_filter.intersection( + set(row.link_to_tags or []) + ) helper.add_candidates(candidates) if initial_roots: diff --git a/libs/knowledge-store/tests/integration_tests/test_graph_store.py b/libs/knowledge-store/tests/integration_tests/test_graph_store.py index 84b66abce..7cf4efd18 100644 --- a/libs/knowledge-store/tests/integration_tests/test_graph_store.py +++ b/libs/knowledge-store/tests/integration_tests/test_graph_store.py @@ -211,10 +211,14 @@ def test_mmr_traversal( results = gs.mmr_traversal_search("0.0", fetch_k=2, k=4, initial_roots=["v0"]) assert _result_ids(results) == ["v1", "v3", "v2"] - results = gs.mmr_traversal_search("0.0", k=2, fetch_k=2, tag_filter=set(("explicit", "link"))) + results = gs.mmr_traversal_search( + "0.0", k=2, fetch_k=2, tag_filter={("explicit", "link")} + ) assert _result_ids(results) == ["v0", "v2"] - results = gs.mmr_traversal_search("0.0", k=2, fetch_k=2, tag_filter=set(("no", "match"))) + results = gs.mmr_traversal_search( + "0.0", k=2, fetch_k=2, tag_filter={("no", "match")} + ) assert _result_ids(results) == []