Skip to content

fix: Localize BigQuery log suppression for gbq.py #2541

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.

Sign up for GitHub

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

Merged
shuoweil merged 8 commits into from
Mar 31, 2026
Merged

fix: Localize BigQuery log suppression for gbq.py #2541

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
a7a3f8c
fix: Localize BigQuery log suppression for gbq.py
shuoweil Mar 27, 2026
d195f07
Merge branch 'main' into shuowei-fix-docs
shuoweil Mar 27, 2026
f0422f3
Revert "fix: Localize BigQuery log suppression for gbq.py"
shuoweil Mar 27, 2026
c9471c7
fix: fix gbq.py doctest locally
shuoweil Mar 27, 2026
ed4be31
fix: fix more doctests with print trick
shuoweil Mar 28, 2026
a30a5cd
fix(doc): suppress BigQuery logs in sim_join doctest
shuoweil Mar 28, 2026
31cb558
Merge branch 'main' into shuowei-fix-docs
shuoweil Mar 30, 2026
bd2f93d
Merge branch 'main' into shuowei-fix-docs
shuoweil Mar 31, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 5 additions & 2 deletions bigframes/operations/ai.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -612,8 +612,11 @@ def sim_join(
>>> df1 = bpd.DataFrame({'animal': ['monkey', 'spider']})
>>> df2 = bpd.DataFrame({'animal': ['scorpion', 'baboon']})

>>> df1.ai.sim_join(df2, left_on='animal', right_on='animal', model=model, top_k=1)
animal animal_1
>>> res = df1.ai.sim_join(df2, left_on='animal', right_on='animal', model=model, top_k=1)
>>> print("---"); print(res) # doctest: +ELLIPSIS
---
...
animal animal_1
0 monkey baboon
1 spider scorpion
<BLANKLINE>
Expand Down
6 changes: 4 additions & 2 deletions bigframes/session/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -664,9 +664,11 @@ def read_gbq_query(
... WHERE year = 2016
... GROUP BY pitcherFirstName, pitcherLastName
... ''', index_col="rowindex")
>>> df.head(2)
>>> print("START_OF_OUTPUT"); df.head(2) # doctest: +ELLIPSIS,+NORMALIZE_WHITESPACE
START_OF_OUTPUT
...
pitcherFirstName pitcherLastName averagePitchSpeed
rowindex
...
1 Albertin Chapman 96.514113
2 Zachary Britton 94.591039
<BLANKLINE>
Expand Down
6 changes: 4 additions & 2 deletions third_party/bigframes_vendored/pandas/io/gbq.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -86,9 +86,11 @@ def read_gbq(
... WHERE year = 2016
... GROUP BY pitcherFirstName, pitcherLastName
... ''', index_col="rowindex")
>>> df.head(2)
>>> print("START_OF_OUTPUT"); df.head(2) # doctest: +ELLIPSIS,+NORMALIZE_WHITESPACE
START_OF_OUTPUT
...
pitcherFirstName pitcherLastName averagePitchSpeed
rowindex
...
1 Albertin Chapman 96.514113
2 Zachary Britton 94.591039
<BLANKLINE>
Expand Down
Loading