From 6dc61b8c1616b28707a83d9cd15fafa4fa3f1382 Mon Sep 17 00:00:00 2001 From: Eric Pinzur Date: Thu, 10 Oct 2024 14:37:08 +0200 Subject: [PATCH] lint --- libs/astradb/langchain_astradb/graph_vectorstores.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/astradb/langchain_astradb/graph_vectorstores.py b/libs/astradb/langchain_astradb/graph_vectorstores.py index 7fc32bf..501b1ea 100644 --- a/libs/astradb/langchain_astradb/graph_vectorstores.py +++ b/libs/astradb/langchain_astradb/graph_vectorstores.py @@ -628,7 +628,7 @@ async def fetch_initial_candidates() -> None: retrieved_docs[doc_id] = doc if doc_id not in outgoing_links_map: - node = self._doc_to_node(doc) + node = _doc_to_node(doc) outgoing_links_map[doc_id] = _outgoing_links(node=node) candidates[doc_id] = embedding helper.add_candidates(candidates) @@ -986,7 +986,7 @@ async def _get_outgoing_links(self, source_ids: Iterable[str]) -> set[Link]: for doc in docs: if doc is not None: - node = self._doc_to_node(doc=doc) + node = _doc_to_node(doc=doc) links.update(_outgoing_links(node=node)) return links