pytest fixes

raghavdixit99 · Apr 13, 2024 · d0d49da · d0d49da
1 parent dad8aa3
commit d0d49da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llama-index-legacy/llama_index/legacy/vector_stores/lancedb.py b/llama-index-legacy/llama_index/legacy/vector_stores/lancedb.py
@@ -14,8 +14,8 @@
     TextNode,
 )
 from llama_index.legacy.vector_stores.types import (
-    BasePydanticVectorStore,
     MetadataFilters,
+    VectorStore,
     VectorStoreQuery,
     VectorStoreQueryResult,
 )
@@ -53,7 +53,7 @@ def _to_llama_similarities(results: DataFrame) -> List[float]:
     return normalized_similarities.tolist()
 
 
-class LanceDBVectorStore(BasePydanticVectorStore):
+class LanceDBVectorStore(VectorStore):
     """
     The LanceDB Vector Store.