maybe fix test

holoviz · Jan 6, 2025 · e31b0d5 · e31b0d5
1 parent bbfca3d
commit e31b0d5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lumen/tests/ai/test_vector_store.py b/lumen/tests/ai/test_vector_store.py
@@ -1,9 +1,15 @@
 import pytest
 
+try:
+    import lumen.ai  # noqa
+except ModuleNotFoundError:
+    pytest.skip("lumen.ai could not be imported, skipping tests.", allow_module_level=True)
+
 from lumen.ai.embeddings import NumpyEmbeddings
 from lumen.ai.vector_store import DuckDBVectorStore, NumpyVectorStore
 
 
+@pytest.mark.xdist_group("vss")
 class VectorStoreTestKit:
     """
     A base class (test kit) that provides the *common* tests and fixture definitions.