Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEAT] [Scan Operator] Add ChunkSpec for specifying format-specific per-file row subset selection for ScanTasks. #1590

Merged
merged 3 commits into from
Nov 11, 2023

Conversation

clarkzinzow
Copy link
Contributor

@clarkzinzow clarkzinzow commented Nov 10, 2023

This PR adds a ChunkSpec abstraction + integration that allows for format-specific per-file row subset selection for ScanTasks. The first concrete implementation is for Parquet, via an index-based row group selection. Pushing this down to be a ScanTask concept makes it easier to specify things like this at a per-file level without polluting user-facing abstractions like the file format config, and should make merging/splitting ScanTasks along the row dimension easier.

In the future, other formats such as CSV and NDJSON could support selecting by byte range ending at row boundaries.

@github-actions github-actions bot added the enhancement New feature or request label Nov 10, 2023
Copy link

codecov bot commented Nov 10, 2023

Codecov Report

Merging #1590 (3c858f8) into main (cdc1b94) will not change coverage.
The diff coverage is n/a.

Additional details and impacted files

Impacted file tree graph

@@           Coverage Diff           @@
##             main    #1590   +/-   ##
=======================================
  Coverage   85.19%   85.19%           
=======================================
  Files          54       54           
  Lines        5180     5180           
=======================================
  Hits         4413     4413           
  Misses        767      767           
Files Coverage Δ
daft/table/micropartition.py 89.58% <ø> (ø)
daft/table/table.py 81.97% <ø> (ø)

src/daft-micropartition/src/micropartition.rs Outdated Show resolved Hide resolved
@clarkzinzow clarkzinzow force-pushed the clark/row-group-selector-per-file branch from d9a32d3 to af75f86 Compare November 10, 2023 19:38
@clarkzinzow clarkzinzow force-pushed the clark/row-group-selector-per-file branch from af75f86 to 3c858f8 Compare November 10, 2023 23:15
@clarkzinzow clarkzinzow merged commit ef4d2fd into main Nov 11, 2023
37 checks passed
@clarkzinzow clarkzinzow deleted the clark/row-group-selector-per-file branch November 11, 2023 00:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants