Minor change

kineticadb · Dec 7, 2023 · 2f92168 · 2f92168
1 parent 19b682c
commit 2f92168
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/langchain/tests/integration_tests/vectorstores/test_kinetica.py b/libs/langchain/tests/integration_tests/vectorstores/test_kinetica.py
@@ -200,7 +200,7 @@ def test_kinetica_relevance_score(get_db: GPUdb) -> None:
     ]
 
 
-@pytest.mark.requires(OpenAIEmbeddings)
+@pytest.mark.requires("openai")
 def test_kinetica_max_marginal_relevance_search(get_db: GPUdb) -> None:
     """Test end to end construction and search."""
     db = get_db
@@ -238,7 +238,7 @@ def test_kinetica_max_marginal_relevance_search_with_score(get_db: GPUdb) -> Non
     assert output == [(Document(page_content="foo"), 0.0)]
 
 
-@pytest.mark.requires(OpenAIEmbeddings)
+@pytest.mark.requires("openai")
 def test_kinetica_with_openai_embeddings(get_db: GPUdb) -> None:
     """Test end to end construction and search."""
     db = get_db