diff --git a/src/datasets/packaged_modules/arrow/arrow.py b/src/datasets/packaged_modules/arrow/arrow.py index a9786745145..bcf31c473d2 100644 --- a/src/datasets/packaged_modules/arrow/arrow.py +++ b/src/datasets/packaged_modules/arrow/arrow.py @@ -45,7 +45,7 @@ def _split_generators(self, dl_manager): with open(file, "rb") as f: try: reader = pa.ipc.open_stream(f) - except pa.lib.ArrowInvalid: + except (OSError, pa.lib.ArrowInvalid): reader = pa.ipc.open_file(f) self.info.features = datasets.Features.from_arrow_schema(reader.schema) break