5
5
import tarfile
6
6
import warnings
7
7
import zipfile
8
- from typing import Any , cast , IO , Iterator , NamedTuple , Optional , Sequence , Tuple
8
+ from typing import Any , cast , Iterator , NamedTuple , Optional , Sequence , Tuple , BinaryIO
9
9
10
10
import numpy
11
11
import torch
@@ -590,7 +590,7 @@ def _open_inside_zip(self, archive_path: str, member_path: Optional[str] = None)
590
590
members_list = archive .namelist ()
591
591
member_path = self ._get_the_only_file_in_the_archive (members_list , archive_path )
592
592
member_path = cast (str , member_path )
593
- member_file = archive .open (member_path , "r" )
593
+ member_file = cast ( BinaryIO , archive .open (member_path , "r" ) )
594
594
self ._handle = io .TextIOWrapper (member_file , encoding = self ._encoding )
595
595
self ._archive_handle = archive
596
596
@@ -602,7 +602,7 @@ def _open_inside_tar(self, archive_path: str, member_path: Optional[str] = None)
602
602
member_path = self ._get_the_only_file_in_the_archive (members_list , archive_path )
603
603
member_path = cast (str , member_path )
604
604
member = archive .getmember (member_path ) # raises exception if not present
605
- member_file = cast (IO [ bytes ] , archive .extractfile (member ))
605
+ member_file = cast (BinaryIO , archive .extractfile (member ))
606
606
self ._handle = io .TextIOWrapper (member_file , encoding = self ._encoding )
607
607
self ._archive_handle = archive
608
608
0 commit comments