LoadDataSet pyarrow.lib.ArrowCapacityError

How about trying .shard()?