Improved code [skip ci]

Author: ankane

pgvector/bit.py

@@ -60,8 +60,8 @@ def __eq__(self, other: object) -> bool:
         return False
 
     def _length(self) -> int:
-        length, = unpack_from('>i', self._value)
-        return cast(int, length)
+        length, = cast(tuple[int], unpack_from('>i', self._value))
+        return length
 
     def to_list(self) -> list[bool]:
         # TODO improve

pgvector/halfvec.py

@@ -38,8 +38,8 @@ def __eq__(self, other: object) -> bool:
         return False
 
     def dimensions(self) -> int:
-        dim, = struct.unpack_from('>H', self._value)
-        return cast(int, dim)
+        dim, = cast(tuple[int], struct.unpack_from('>H', self._value))
+        return dim
 
     def to_list(self) -> list[float]:
         return list(struct.unpack_from(f'>{self.dimensions()}e', self._value[4:]))

pgvector/vector.py

@@ -38,8 +38,8 @@ def __eq__(self, other: object) -> bool:
         return False
 
     def dimensions(self) -> int:
-        dim, = struct.unpack_from('>H', self._value)
-        return cast(int, dim)
+        dim, = cast(tuple[int], struct.unpack_from('>H', self._value))
+        return dim
 
     def to_list(self) -> list[float]:
         return list(struct.unpack_from(f'>{self.dimensions()}f', self._value[4:]))