This commit is contained in:
Adam J. Stewart 2021-06-09 21:38:52 +00:00
Родитель a585a94c04
Коммит 36611a7c6a
1 изменённых файлов: 3 добавлений и 3 удалений

Просмотреть файл

@ -186,10 +186,10 @@ class _COWC(VisionDataset, abc.ABC):
md5=md5,
)
if filename.endswith(".tbz"):
with tarfile.TarFile(
with tarfile.open(
os.path.join(self.root, self.base_folder, filename)
) as f:
f.extractall(os.path.join(self.root, self.base_folder))
) as tar:
tar.extractall(os.path.join(self.root, self.base_folder))
elif filename.endswith(".bz2"):
filepath = os.path.join(self.root, self.base_folder, filename)
with bz2.BZ2File(filepath) as old_fh: