Improved typing [skip ci]

Author: ankane

pgvector/bit.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 from struct import pack, unpack_from
+from typing import cast
 
 try:
     import numpy as np
@@ -60,7 +61,7 @@ def __eq__(self, other: object) -> bool:
 
     def _length(self) -> int:
         length, = unpack_from('>i', self._value)
-        return length
+        return cast(int, length)
 
     def to_list(self) -> list[bool]:
         # TODO improve

pgvector/halfvec.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 import struct
+from typing import cast
 
 try:
     import numpy as np
@@ -38,7 +39,7 @@ def __eq__(self, other: object) -> bool:
 
     def dimensions(self) -> int:
         dim, = struct.unpack_from('>H', self._value)
-        return dim
+        return cast(int, dim)
 
     def to_list(self) -> list[float]:
         return list(struct.unpack_from(f'>{self.dimensions()}e', self._value[4:]))

pgvector/vector.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 import struct
+from typing import cast
 
 try:
     import numpy as np
@@ -38,7 +39,7 @@ def __eq__(self, other: object) -> bool:
 
     def dimensions(self) -> int:
         dim, = struct.unpack_from('>H', self._value)
-        return dim
+        return cast(int, dim)
 
     def to_list(self) -> list[float]:
         return list(struct.unpack_from(f'>{self.dimensions()}f', self._value[4:]))

pyproject.toml

@@ -46,8 +46,9 @@ files = [
     "tests"
 ]
 strict = true
-disable_error_code = ["unused-ignore"]
+disable_error_code = ["unused-ignore", "no-untyped-def", "no-untyped-call"]
 ignore_missing_imports = true
+disallow_subclassing_any = false
 python_executable = ".venv/bin/python"
 
 [tool.pyright]