Added overloads [skip ci]

Author: ankane

pgvector/sparsevec.py

@@ -1,12 +1,24 @@
 from __future__ import annotations
 import numpy as np
 from struct import pack, unpack_from
-from typing import Any
+from typing import Any, overload
 
 NO_DEFAULT = object()
 
 
 class SparseVector:
+    @overload
+    def __init__(self, value: dict[int, float], dimensions: int, /) -> None:
+        ...
+
+    @overload
+    def __init__(self, value: list[float], /) -> None:
+        ...
+
+    @overload
+    def __init__(self, value: Any, /) -> None:
+        ...
+
     def __init__(self, value: dict[int, float] | list[float] | Any, dimensions: int | Any = NO_DEFAULT, /) -> None:
         if value.__class__.__module__.startswith('scipy.sparse.'):
             if dimensions is not NO_DEFAULT:

tests/test_sparse_vector.py

@@ -14,7 +14,7 @@ def test_list(self):
 
     def test_list_dimensions(self):
         with pytest.raises(ValueError) as error:
-            SparseVector([1, 0, 2, 0, 3, 0], 6)
+            SparseVector([1, 0, 2, 0, 3, 0], 6)  # ty: ignore[invalid-argument-type]
         assert str(error.value) == 'extra argument'
 
     def test_ndarray(self):
@@ -40,7 +40,7 @@ def test_coo_array(self):
 
     def test_coo_array_dimensions(self):
         with pytest.raises(ValueError) as error:
-            SparseVector(coo_array(np.array([1, 0, 2, 0, 3, 0])), 6)
+            SparseVector(coo_array(np.array([1, 0, 2, 0, 3, 0])), 6)  # ty: ignore[invalid-argument-type]
         assert str(error.value) == 'extra argument'
 
     def test_coo_matrix(self):