Skip to content
Snippets Groups Projects
Commit c65a3167 authored by Jan Ebert's avatar Jan Ebert
Browse files

Support Arrow reader not requiring closing

parent f9c6cff6
Branches master
No related tags found
No related merge requests found
......@@ -60,7 +60,7 @@ class FixlengthArrow(torch.utils.data.Dataset):
return img, target
def __del__(self):
if self.reader is not None:
if self.reader is not None and hasattr(self.reader, 'close'):
self.reader.close()
if self.arrowfile is not None:
self.arrowfile.close()
......
......
......@@ -59,7 +59,7 @@ class VarlengthArrow(torch.utils.data.Dataset):
return img, target
def __del__(self):
if self.reader is not None:
if self.reader is not None and hasattr(self.reader, 'close'):
self.reader.close()
if self.arrowfile is not None:
self.arrowfile.close()
......
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment