|
9 | 9 | from array import array |
10 | 10 | from binascii import hexlify, unhexlify |
11 | 11 | from collections.abc import ByteString |
12 | | -from io import BytesIO, IOBase |
| 12 | +from io import BufferedRandom, BytesIO, IOBase |
13 | 13 | from mmap import mmap |
14 | 14 | from numbers import Number |
15 | 15 | from pathlib import Path |
@@ -605,7 +605,7 @@ def __init__( |
605 | 605 | ) -> None: |
606 | 606 | # set some things up |
607 | 607 | self._filepath = resolve_path(filepath) |
608 | | - self.__file_pointer = None |
| 608 | + self.__file_pointer: Union[BufferedRandom, None] = None |
609 | 609 | super().__init__(est_elements, false_positive_rate, filepath, hex_string, hash_function) |
610 | 610 |
|
611 | 611 | def _load_init(self, filepath, hash_function, hex_string, est_elements, false_positive_rate): |
@@ -640,7 +640,7 @@ def close(self) -> None: |
640 | 640 | """Clean up the BloomFilterOnDisk object""" |
641 | 641 | if self.__file_pointer is not None and not self.__file_pointer.closed: |
642 | 642 | self.__update() |
643 | | - self._bloom.close() |
| 643 | + self._bloom.close() # type: ignore |
644 | 644 | self.__file_pointer.close() |
645 | 645 | self.__file_pointer = None |
646 | 646 |
|
@@ -669,7 +669,7 @@ def _load(self, file: Union[str, Path], hash_function: Union[HashFuncT, None] = |
669 | 669 | fpr, n_hashes, n_bits = self._get_optimized_params(est_els, fpr) |
670 | 670 | self._set_values(est_els, fpr, n_hashes, n_bits, hash_function) |
671 | 671 | # setup a few additional items |
672 | | - self.__file_pointer = open(file, "r+b") # type: ignore # noqa: SIM115 |
| 672 | + self.__file_pointer = open(file, "r+b") # noqa: SIM115 |
673 | 673 | self._bloom = mmap(self.__file_pointer.fileno(), 0) # type: ignore |
674 | 674 | self._on_disk = True |
675 | 675 |
|
|
0 commit comments