fix extra tests

datastax · Feb 5, 2025 · 5313962 · 5313962
1 parent 017cf4e
commit 5313962
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 6 deletions.
diff --git a/packages/langchain-graph-retriever/tests/transformers/test_gliner.py b/packages/langchain-graph-retriever/tests/transformers/test_gliner.py
@@ -2,12 +2,12 @@
 
 import pytest
 from langchain_core.documents import Document
-from langchain_graph_retriever.transformers.gliner import GLiNERTransformer
 
 
 @pytest.mark.extra
 def test_transform_documents(animal_docs: list[Document]):
     from gliner import GLiNER  # type: ignore
+    from langchain_graph_retriever.transformers.gliner import GLiNERTransformer
 
     class FakeGLiNER(GLiNER):
         def __init__(self):

diff --git a/packages/langchain-graph-retriever/tests/transformers/test_html_hyperlink.py b/packages/langchain-graph-retriever/tests/transformers/test_html_hyperlink.py
@@ -1,8 +1,5 @@
 import pytest
 from langchain_core.documents import Document
-from langchain_graph_retriever.transformers.html_hyperlink import (
-    HtmlHyperlinkTransformer,
-)
 
 test_html = """
 <!DOCTYPE html>
@@ -28,6 +25,10 @@
 
 @pytest.mark.extra
 def test_transform_documents():
+    from langchain_graph_retriever.transformers.html_hyperlink import (
+        HtmlHyperlinkTransformer,
+    )
+
     doc = Document(
         id="animal_html",
         page_content=test_html,

diff --git a/packages/langchain-graph-retriever/tests/transformers/test_keybert.py b/packages/langchain-graph-retriever/tests/transformers/test_keybert.py
@@ -2,12 +2,12 @@
 
 import pytest
 from langchain_core.documents import Document
-from langchain_graph_retriever.transformers.keybert import KeyBERTTransformer
 
 
 @pytest.mark.extra
 def test_transform_documents(animal_docs: list[Document]):
     from keybert import KeyBERT  # type: ignore
+    from langchain_graph_retriever.transformers.keybert import KeyBERTTransformer
 
     class FakeKeyBERT(KeyBERT):
         def __init__(self):

diff --git a/packages/langchain-graph-retriever/tests/transformers/test_spacy.py b/packages/langchain-graph-retriever/tests/transformers/test_spacy.py
@@ -1,10 +1,10 @@
 import pytest
 from langchain_core.documents import Document
-from langchain_graph_retriever.transformers.spacy import SpacyNERTransformer
 
 
 @pytest.mark.extra
 def test_transform_documents(animal_docs: list[Document]):
+    from langchain_graph_retriever.transformers.spacy import SpacyNERTransformer
     from spacy.language import Language
     from spacy.tokens import Doc, Span
     from spacy.vocab import Vocab