fix:vectordb lazy load

This commit is contained in:
aries_ckt
2023-10-17 11:52:45 +08:00
parent 88f1daaa50
commit 496696537d
6 changed files with 47 additions and 36 deletions

View File

@@ -3,8 +3,9 @@ import pytest
from pilot import vector_store
from pilot.vector_store.base import VectorStoreBase
def test_vetorestore_imports() -> None:
""" Simple test to make sure all things can be imported."""
for cls in vector_store.__all__:
def test_vetorestore_imports() -> None:
"""Simple test to make sure all things can be imported."""
for cls in vector_store.__all__:
assert issubclass(getattr(vector_store, cls), VectorStoreBase)