Switched to positional-only parameters

Author: ankane

pgvector/_utils.py

@@ -11,13 +11,13 @@
     ndarray = None
 
 
-def is_ndarray(value: object) -> bool:
+def is_ndarray(value: object, /) -> bool:
     if (numpy := sys.modules.get('numpy')):
         return isinstance(value, numpy.ndarray)
     return False
 
 
-def is_sparse_array(value: object) -> bool:
+def is_sparse_array(value: object, /) -> bool:
     if (sparse := sys.modules.get('scipy.sparse')):
         return isinstance(value, (sparse.sparray, sparse.spmatrix))
     return False

pgvector/bit.py

@@ -11,7 +11,7 @@ class Bit:
     _length: int
     _data: bytes
 
-    def __init__(self, value: bytes | str | list[bool] | np.ndarray[tuple[int, ...], np.dtype[np.bool | np.uint8]]) -> None:
+    def __init__(self, value: bytes | str | list[bool] | np.ndarray[tuple[int, ...], np.dtype[np.bool | np.uint8]], /) -> None:
         if isinstance(value, bytes):
             self._length = 8 * len(value)
             self._data = value
@@ -52,7 +52,7 @@ def __init__(self, value: bytes | str | list[bool] | np.ndarray[tuple[int, ...],
     def __repr__(self) -> str:
         return f'Bit({self.to_text()})'
 
-    def __eq__(self, other: object) -> bool:
+    def __eq__(self, other: object, /) -> bool:
         if isinstance(other, self.__class__):
             return self._length == other._length and self._data == other._data
         return False
@@ -73,12 +73,12 @@ def to_binary(self) -> bytes:
         return pack('>i', self._length) + self._data
 
     @classmethod
-    def from_text(cls, value: str) -> Bit:
+    def from_text(cls, value: str, /) -> Bit:
         # cast to ensure always uses str constructor
         return cls(str(value))
 
     @classmethod
-    def from_binary(cls, value: bytes | bytearray | memoryview) -> Bit:
+    def from_binary(cls, value: bytes | bytearray | memoryview, /) -> Bit:
         length, = unpack_from('>i', value)
         data = memoryview(value)[4:].tobytes()
 

pgvector/halfvec.py

@@ -12,7 +12,7 @@
 class HalfVector:
     _value: array[int]  # uses uint16 since no float16
 
-    def __init__(self, value: list[float] | ndarray) -> None:
+    def __init__(self, value: list[float] | ndarray, /) -> None:
         if isinstance(value, list):
             dim = len(value)
             try:
@@ -34,7 +34,7 @@ def __init__(self, value: list[float] | ndarray) -> None:
     def __repr__(self) -> str:
         return f'HalfVector({self.to_list()})'
 
-    def __eq__(self, other: object) -> bool:
+    def __eq__(self, other: object, /) -> bool:
         if isinstance(other, self.__class__):
             return self._value == other._value
         return False
@@ -62,11 +62,11 @@ def to_binary(self) -> bytes:
         return struct.pack('>HH', len(value), 0) + memoryview(value)
 
     @classmethod
-    def from_text(cls, value: str) -> HalfVector:
+    def from_text(cls, value: str, /) -> HalfVector:
         return cls(cls._from_text(value))
 
     @classmethod
-    def from_binary(cls, value: bytes | bytearray | memoryview) -> HalfVector:
+    def from_binary(cls, value: bytes | bytearray | memoryview, /) -> HalfVector:
         dim, unused = struct.unpack_from('>HH', value)
         data = memoryview(value)[4:]
 
@@ -86,11 +86,11 @@ def from_binary(cls, value: bytes | bytearray | memoryview) -> HalfVector:
         return vec
 
     @classmethod
-    def _from_text(cls, value: str) -> list[float]:
+    def _from_text(cls, value: str, /) -> list[float]:
         return [float(v) for v in value[1:-1].split(',')]
 
     @classmethod
-    def _to_db(cls, value: list[float] | ndarray | HalfVector | None) -> str | None:
+    def _to_db(cls, value: list[float] | ndarray | HalfVector | None, /) -> str | None:
         if value is None:
             return value
 
@@ -104,7 +104,7 @@ def _to_db(cls, value: list[float] | ndarray | HalfVector | None) -> str | None:
         return value.to_text()
 
     @classmethod
-    def _from_db(cls, value: str | HalfVector | None) -> list[float] | None:
+    def _from_db(cls, value: str | HalfVector | None, /) -> list[float] | None:
         if value is None:
             return value
 

pgvector/sparsevec.py

@@ -52,7 +52,7 @@ def __repr__(self) -> str:
         elements = dict(zip(self._indices, self._values))
         return f'SparseVector({elements}, {self._dim})'
 
-    def __eq__(self, other: object) -> bool:
+    def __eq__(self, other: object, /) -> bool:
         if isinstance(other, self.__class__):
             return self._dim == other._dim and self._indices == other._indices and self._values == other._values
         return False
@@ -102,7 +102,7 @@ def _from_dict(self, d: dict[int, float], dim: int) -> None:
         self._indices = [int(v[0]) for v in elements]
         self._values = [float(v[1]) for v in elements]
 
-    def _from_sparse(self, arr: sparray | spmatrix) -> None:
+    def _from_sparse(self, arr: sparray | spmatrix, /) -> None:
         value: coo_array | coo_matrix = arr.tocoo(copy=False)  # type: ignore
 
         shape = cast(tuple[int], value.shape)
@@ -120,13 +120,13 @@ def _from_sparse(self, arr: sparray | spmatrix) -> None:
             self._indices = value.col.tolist()
         self._values = value.data.tolist()
 
-    def _from_dense(self, value: list[float] | ndarray) -> None:
+    def _from_dense(self, value: list[float] | ndarray, /) -> None:
         self._dim = len(value)
         self._indices = [i for i, v in enumerate(value) if v != 0]
         self._values = [float(value[i]) for i in self._indices]
 
     @classmethod
-    def from_text(cls, value: str) -> SparseVector:
+    def from_text(cls, value: str, /) -> SparseVector:
         elements, dim = value.split('/', 2)
         indices: list[int] = []
         values: list[float] = []
@@ -139,7 +139,7 @@ def from_text(cls, value: str) -> SparseVector:
         return cls._from_parts(int(dim), indices, values)
 
     @classmethod
-    def from_binary(cls, value: bytes | bytearray | memoryview) -> SparseVector:
+    def from_binary(cls, value: bytes | bytearray | memoryview, /) -> SparseVector:
         dim, nnz, unused = unpack_from('>iii', value)
 
         if len(value) != 12 + 8 * nnz:
@@ -153,15 +153,15 @@ def from_binary(cls, value: bytes | bytearray | memoryview) -> SparseVector:
         return cls._from_parts(dim, indices, values)
 
     @classmethod
-    def _from_parts(cls, dim: int, indices: list[int], values: list[float]) -> SparseVector:
+    def _from_parts(cls, dim: int, indices: list[int], values: list[float], /) -> SparseVector:
         vec = cls.__new__(cls)
         vec._dim = dim
         vec._indices = indices
         vec._values = values
         return vec
 
     @classmethod
-    def _to_db(cls, value: list[float] | ndarray | sparray | spmatrix | SparseVector | None) -> str | None:
+    def _to_db(cls, value: list[float] | ndarray | sparray | spmatrix | SparseVector | None, /) -> str | None:
         if value is None:
             return value
 
@@ -171,7 +171,7 @@ def _to_db(cls, value: list[float] | ndarray | sparray | spmatrix | SparseVector
         return value.to_text()
 
     @classmethod
-    def _from_db(cls, value: str | SparseVector | None) -> SparseVector | None:
+    def _from_db(cls, value: str | SparseVector | None, /) -> SparseVector | None:
         if value is None or isinstance(value, SparseVector):
             return value
 

pgvector/vector.py

@@ -12,7 +12,7 @@
 class Vector:
     _value: array[float]
 
-    def __init__(self, value: list[float] | ndarray) -> None:
+    def __init__(self, value: list[float] | ndarray, /) -> None:
         if isinstance(value, list):
             try:
                 self._value = array('f', value)
@@ -33,7 +33,7 @@ def __init__(self, value: list[float] | ndarray) -> None:
     def __repr__(self) -> str:
         return f'Vector({self.to_list()})'
 
-    def __eq__(self, other: object) -> bool:
+    def __eq__(self, other: object, /) -> bool:
         if isinstance(other, self.__class__):
             return self._value == other._value
         return False
@@ -60,11 +60,11 @@ def to_binary(self) -> bytes:
         return struct.pack('>HH', len(value), 0) + memoryview(value)
 
     @classmethod
-    def from_text(cls, value: str) -> Vector:
+    def from_text(cls, value: str, /) -> Vector:
         return cls(cls._from_text(value))
 
     @classmethod
-    def from_binary(cls, value: bytes | bytearray | memoryview) -> Vector:
+    def from_binary(cls, value: bytes | bytearray | memoryview, /) -> Vector:
         dim, unused = struct.unpack_from('>HH', value)
         data = memoryview(value)[4:]
 
@@ -84,11 +84,11 @@ def from_binary(cls, value: bytes | bytearray | memoryview) -> Vector:
         return vec
 
     @classmethod
-    def _from_text(cls, value: str) -> list[float]:
+    def _from_text(cls, value: str, /) -> list[float]:
         return [float(v) for v in value[1:-1].split(',')]
 
     @classmethod
-    def _to_db(cls, value: list[float] | ndarray | Vector | None) -> str | None:
+    def _to_db(cls, value: list[float] | ndarray | Vector | None, /) -> str | None:
         if value is None:
             return value
 
@@ -102,7 +102,7 @@ def _to_db(cls, value: list[float] | ndarray | Vector | None) -> str | None:
         return value.to_text()
 
     @classmethod
-    def _from_db(cls, value: str | Vector | None) -> list[float] | None:
+    def _from_db(cls, value: str | Vector | None, /) -> list[float] | None:
         if value is None:
             return value