Skip to content
This repository was archived by the owner on May 29, 2026. It is now read-only.

feat: asset partition row count - #11

Merged
aaaaahaaaaa merged 1 commit into
mainfrom
feat/partition_row_count
Mar 5, 2026
Merged

feat: asset partition row count#11
aaaaahaaaaa merged 1 commit into
mainfrom
feat/partition_row_count

Conversation

@aaaaahaaaaa

@aaaaahaaaaa aaaaahaaaaa commented Mar 5, 2026

Copy link
Copy Markdown
Collaborator

Note

Medium Risk
Introduces new abstract methods on core IO interfaces and new SQL/BigQuery query paths, which can break custom IO implementations and may surface backend-specific edge cases in grouping/count semantics.

Overview
Adds an introspection API to compute per-partition row counts: IO.partition_row_counts() (new abstract method), DatabaseIO._count_by_partition()/partition_row_counts(), and concrete implementations for SqlIO, BigQueryIO, FileIO, and MemoryIO.

Exposes the feature at the asset level via Asset.partition_row_counts() (including multi-IO key resolution), adds a DAG.backfill() convenience wrapper, updates the Postgres example to use time-partitioned backfills and print counts, and introduces/updates tests covering the new behavior and error cases (non-partitioned assets, missing tables, invalid IO keys).

Written by Cursor Bugbot for commit b5ff353. This will update automatically on new commits. Configure here.

@aaaaahaaaaa aaaaahaaaaa self-assigned this Mar 5, 2026
@aaaaahaaaaa
aaaaahaaaaa merged commit 2867b91 into main Mar 5, 2026
5 of 6 checks passed
@aaaaahaaaaa
aaaaahaaaaa deleted the feat/partition_row_count branch March 5, 2026 14:51
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant