fix tests

datastax · Feb 5, 2025 · 94da13b · 94da13b
1 parent e7ee9ec
commit 94da13b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -70,7 +70,7 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Install Python dependencies
-        run: uv sync --frozen --all-packages
+        run: uv sync --frozen --all-packages --all-extras
         shell: bash
 
       - name: Test graph-retriever

diff --git a/packages/langchain-graph-retriever/tests/transformers/test_spacy.py b/packages/langchain-graph-retriever/tests/transformers/test_spacy.py
@@ -33,7 +33,7 @@ def test_transform_documents(animal_docs: list[Document]):
     assert "third: is" in transformed_docs[0].metadata["spacey"]
 
     transformer = SpacyNERTransformer(
-        model=fake_model, metadata_key="spacey", include_labels=set("first")
+        model=fake_model, metadata_key="spacey", include_labels=set(["first"])
     )
     transformed_docs = transformer.transform_documents(animal_docs)
     assert "spacey" in transformed_docs[0].metadata
@@ -42,7 +42,7 @@ def test_transform_documents(animal_docs: list[Document]):
     assert "third: is" not in transformed_docs[0].metadata["spacey"]
 
     transformer = SpacyNERTransformer(
-        model=fake_model, metadata_key="spacey", exclude_labels=set("first")
+        model=fake_model, metadata_key="spacey", exclude_labels=set(["first"])
     )
     transformed_docs = transformer.transform_documents(animal_docs)
     assert "spacey" in transformed_docs[0].metadata

diff --git a/uv.lock b/uv.lock