perf: materialize sections once in resolve_batch and trim before resolution - #110
Merged
Conversation
…lution resolve_batch re-executed the entire upstream lazy plan (source scan, encoding/regex ops, NLP normalization) once per resolved node column: distinct().collect() per column, join_matches' collect per column, and log_unmatched's anti-join collect with --log. A section resolving subject + object + two qualifiers ran that pipeline ~5x (~9x logging). ### Implementation - resolve_batch collects the frame exactly once; term extraction, unmatched logging, and per-column join-backs run on the in-memory frame. - join_matches keeps its LazyFrame contract over a new eager core (_join_matches_eager) that resolve_batch calls directly. - trim moves from the finalize ops to just before resolve_batch, so the materialized frame and downstream joins drop the spent raw column_<n> columns early; its phase label falls back to transform. ### Testing - uv run pytest -q: 928 passed (1 failure + 1 collection error are pre-existing: this venv lacks the [qc] extra; both fail on the clean tree too). - ruff check / ruff format --check / pyright: clean. - Synthetic 50k-row, 4-column benchmark: resolve phase 0.14s -> 0.08s (~1.75x), identical output rows.
|
Important
This repository does not receive automatic reviews because it has fewer than 10 stars. ⚙️ Run configurationConfiguration used: defaults Review profile: CHILL Plan: Pro Plus Run ID: Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out. Comment |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
resolve_batchre-executed the entire upstream lazy plan — source scan, encoding/regex ops, NLP normalization — once per resolved node column:distinct(...).collect()per column,join_matches' collect per column, andlog_unmatched's anti-join collect with--log. A section resolving subject + object + two qualifiers ran that pipeline ~5× (~9× with logging).Changes
resolve_batch(fullmap.py) collects the frame exactly once; per-column term extraction, unmatched logging, and join-backs all run against the in-memory frame. Output rows are unchanged.join_matcheskeeps its LazyFrame-in/LazyFrame-out contract over a new eager core (_join_matches_eager) thatresolve_batchcalls directly.trimmoves from the finalize ops (_provenance_ops) to just beforeresolve_batch(_node_ops), so the materialized frame and every downstream join drop the spent rawcolumn_<n>columns early. All readers of those columns run earlier, so this is safe;prune_to_classskips them regardless via itsnot any(accepts)branch.trim'sPHASE_OFentry is removed so progress falls back totransforminstead of flashingfinalizebeforeresolve.No reordering of filters was needed — every row-reducing op already runs before resolution (pinned by
tests/test_lib.py).Testing
uv run pytest -q→ 928 passed; the 1 failure + 1 collection error are pre-existing environment issues (this venv lacks the[qc]extra — both fail on the clean tree too).ruff check,ruff format --check,pyright→ clean.--log): resolve phase 0.14s → 0.08s (~1.75×), identical output rows. Real tables with heavier encoding chains should gain more, since the eliminated cost is per-column re-execution of the whole upstream plan.Changelog entry added under Unreleased → Performance.