test fix

activeloopai · Aug 15, 2023 · 4d7bfa8 · 4d7bfa8
1 parent 014e09c
commit 4d7bfa8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deeplake/core/vectorstore/test_deeplake_vectorstore.py b/deeplake/core/vectorstore/test_deeplake_vectorstore.py
@@ -1754,7 +1754,7 @@ def test_query_dim(local_path):
     )
 
     vector_store.add(text=texts, embedding=embeddings)
-    with pytest.raises(AssertionError):
+    with pytest.raises(NotImplementedError):
         vector_store.search([texts[0], texts[0]], embedding_fn3, k=1)
 
     vector_store.search([texts[0]], embedding_fn4, k=1)