Garrett Weaver
10/24/2025, 8:17 PMparquet_inflation_factor still respected in flotilla? I have a parquet file that is ~2GB, but expands to ~16GB in memory. When trying to run a select with window function, it is trying to load all data into memory and struggling (OOMing). What is the best course of action here?Garrett Weaver
10/24/2025, 8:20 PMinto_partitions and repartition have similar issues
daft.context.set_execution_config(
parquet_inflation_factor=10,
scan_tasks_min_size_bytes=int(1 * 1024 * 1024),
parquet_target_filesize=int(64 * 1024 * 1024),
parquet_target_row_group_size=int(16 * 1024 * 1024),
)Garrett Weaver
10/24/2025, 8:31 PMtesting = (
daft.read_parquet("~/Desktop/some_data")
.select(
"date",
"id",
row_number()
.over(
Window()
.partition_by(
daft.col("date"),
daft.col("id"),
)
.order_by(daft.col("some_col"), desc=[True])
)
.alias("rn"),
)
.where(daft.col("rn") == daft.lit(1))
.exclude("rn")
)Malcolm Greaves
10/25/2025, 10:43 PMColin Ho
10/27/2025, 5:56 AMColin Ho
10/27/2025, 5:56 AMparquet_inflation_factor still respected in flotilla: Yes, it is respected on a per partition basisColin Ho
10/27/2025, 11:42 AMGarrett Weaver
10/27/2025, 7:41 PMHmmmm unfortunately right now the window functions fully materialize all the data per partition.got it, I think this makes sense....so right now it is treating that whole parquet file as a single partition?
Colin Ho
10/27/2025, 10:07 PM