diff --git a/visidata/loaders/parquet.py b/visidata/loaders/parquet.py index d4fb99c69..50ca020ed 100644 --- a/visidata/loaders/parquet.py +++ b/visidata/loaders/parquet.py @@ -23,8 +23,11 @@ def iterload(self): pq = vd.importExternal("pyarrow.parquet", "pyarrow") from visidata.loaders.arrow import arrow_to_vdtype - with self.source.open('rb') as f: - self.tbl = pq.read_table(f) + if self.source.is_dir(): + self.tbl = pq.read_table(str(self.source)) + else: + with self.source.open('rb') as f: + self.tbl = pq.read_table(f) self.columns = [] for colname, col in zip(self.tbl.column_names, self.tbl.columns):