[ARROW-15286] [Python] pyarrow.dataset.FileSystemDataset.take method causes Segmentation Fault

• Type: Bug
• Status: Resolved
• Priority: Major
• Resolution: Fixed
• Affects Version/s: 5.0.0, 6.0.0, 6.0.1
• Fix Version/s: 7.0.0
• Component/s: Python
• Labels: pull-request-available
• Environment: Ubuntu 18.04, Python 3.8.6; macOS 11.6.2, Python 3.7.5
• External issue URL: https://github.com/apache/arrow/issues/30777
• Assignee: Joris Van den Bossche
• Reporter: Dustin Zubke
• Created: 08/Jan/22 00:26
• Updated: 11/Jan/23 11:36
• Resolved: 10/Jan/22 17:24

Description

Whenever I try calling the `pyarrow.dataset.FileSystemDataset.take` method, I get a segmentation fault. 

I first encountered this using a proprietary dataset and recreated it with the UC Davis data below. I can successfully run the pyarrow.dataset.FileSystemDataset.to_batches method but not the take() method.

Steps to recreate:

!wget https://anson.ucdavis.edu/~clarkf/pems_parquet.zip
!unzip -q pems_parquet.zip

import pyarrow.dataset as ds
file_path= "./pems_sorted/station=402264/part-r-00151-ddaee723-f3f6-4f25-a34b-3312172aa6d7.snappy.parquet"
dataset = ds.dataset(file_path)
dataset.take(1)
>>> 80874 segmentation fault

Creating a dataset from a directory as below also results in a segfault.

dir_path = "./pems_sorted/station=402264"
dataset = ds.dataset(dir_path)
dataset.take(1)

Environments tried:

• Ubuntu 18.04, Python 3.8.6
• macOS 11.6.2, Python 3.7.5

Issue Links

• links to: GitHub Pull Request #12115