Fixed check

pgvector · Jun 26, 2024 · 1451cc6 · 1451cc6
1 parent 0cf5ca5
commit 1451cc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pgvector/utils/sparsevec.py b/pgvector/utils/sparsevec.py
@@ -6,7 +6,7 @@
 
 class SparseVector:
     def __init__(self, value, dimensions=NO_DEFAULT, /):
-        if value.__class__.__module__ == 'scipy.sparse._arrays':
+        if value.__class__.__module__.startswith('scipy.sparse.'):
             if dimensions is not NO_DEFAULT:
                 raise ValueError('dimensions not allowed')