cass

datastax · Mar 13, 2024 · 0403c5e · 0403c5e
1 parent e7a3940
commit 0403c5e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit-tests/test_ragstack.py b/tests/unit-tests/test_ragstack.py
@@ -1,6 +1,6 @@
 def test_import():
     from llama_index.vector_stores.astra_db import AstraDBVectorStore  # noqa
-    from llama_index.vector_stores import CassandraVectorStore  # noqa
+    from llama_index.vector_stores.cassandra import CassandraVectorStore  # noqa
     from langchain.vectorstores import AstraDB  # noqa
     from langchain_astradb import AstraDBVectorStore  # noqa
     import langchain_core  # noqa