Skip to content

Commit aae79a9

Browse files
authored
fix:fix pyre check 3 (#9)
1 parent 8ac91fd commit aae79a9

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

megstore/indexed/msgpack.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def __init__(self, *args, **kwargs):
6262

6363
def _read_array_header(
6464
self,
65-
unpacker: "compat_msgpack.Unpacker", # pytype: disable=attribute-error
65+
unpacker: "compat_msgpack.Unpacker", # type: ignore
6666
) -> int:
6767
try:
6868
return unpacker.read_array_header()
@@ -79,7 +79,7 @@ def _read_array_header(
7979
@classmethod
8080
def _build_index(
8181
cls,
82-
fp_data: BinaryIO,
82+
file_object: BinaryIO,
8383
offsets: Union[IndexHandlerReader, array],
8484
index_build_callback: Optional[Callable[[Any], None]] = None,
8585
) -> int:
@@ -88,8 +88,8 @@ def _build_index(
8888
:raises ValueError: Cannot read valid msgpack array header from msgpack stream
8989
:returns: the latest offset
9090
"""
91-
unpacker = cls._get_msgpack_unpacker(fp_data)
92-
cls._read_array_header(fp_data, unpacker) # pytype: disable=wrong-arg-count
91+
unpacker = cls._get_msgpack_unpacker(file_object)
92+
cls._read_array_header(file_object, unpacker) # type: ignore
9393

9494
# After reading array header, unpacker is at the start byte of the first value
9595
current_offset = unpacker.tell()
@@ -171,7 +171,7 @@ def _batch_get(self, index_slice: slice) -> Iterator[T]:
171171
@classmethod
172172
def _get_msgpack_unpacker(
173173
cls, file_object: BinaryIO, max_buffer_size: Optional[int] = None
174-
) -> "compat_msgpack.Unpacker": # pytype: disable=attribute-error
174+
) -> "compat_msgpack.Unpacker": # type: ignore
175175
"""Get an Unpacker object created with the current instance's msgpack stream
176176
177177
:param max_buffer_size: Maximum buffer size of Unpacker object (bytes),

0 commit comments

Comments
 (0)