Fixed from_sparse method for scipy < 1.13

ankane · ankane · commit d858fcdfaee4 · 2024-06-04T18:21:40.000-07:00
diff --git a/pgvector/utils/sparsevec.py b/pgvector/utils/sparsevec.py
@@ -24,11 +24,18 @@ def from_dict(cls, d, dim):
     def from_sparse(cls, value):
         value = value.tocoo()
 
-        if value.ndim != 1:
+        if value.ndim == 1:
+            dim = value.shape[0]
+        elif value.ndim == 2 and value.shape[0] == 1:
+            dim = value.shape[1]
+        else:
             raise ValueError('expected ndim to be 1')
 
-        dim = value.shape[0]
-        indices = value.coords[0].tolist()
+        if hasattr(value, 'coords'):
+            # scipy 1.13+
+            indices = value.coords[0].tolist()
+        else:
+            indices = value.col.tolist()
         values = value.data.tolist()
         return cls(dim, indices, values)
 
