Removed support for array from from_coordinates [skip ci]

ankane · ankane · commit 08a73cdf7dc0 · 2024-06-04T16:13:47.000-07:00
diff --git a/pgvector/utils/sparsevec.py b/pgvector/utils/sparsevec.py
@@ -14,9 +14,7 @@ def __repr__(self):
 
     @classmethod
     def from_coordinates(cls, coordinates, dim):
-        if isinstance(coordinates, dict):
-            coordinates = coordinates.items()
-        elements = [(i, v) for i, v in coordinates]
+        elements = [(i, v) for i, v in coordinates.items()]
         elements.sort()
         indices = [int(v[0]) for v in elements]
         values = [float(v[1]) for v in elements]
diff --git a/tests/test_sparse_vector.py b/tests/test_sparse_vector.py
@@ -10,7 +10,6 @@ def test_from_dense(self):
 
     def test_from_coordinates(self):
         assert SparseVector.from_coordinates({0: 1, 2: 2, 4: 3}, 6).to_list() == [1, 0, 2, 0, 3, 0]
-        assert SparseVector.from_coordinates([(0, 1), (2, 2), (4, 3)], 6).to_list() == [1, 0, 2, 0, 3, 0]
 
     def test_repr(self):
         assert repr(SparseVector.from_dense([1, 0, 2, 0, 3, 0])) == 'SparseVector(6, [0, 2, 4], [1.0, 2.0, 3.0])'
