Skip to content
This repository was archived by the owner on May 7, 2026. It is now read-only.
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
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
Prev Previous commit
Next Next commit
fix: fix more doctests with print trick
  • Loading branch information
shuoweil committed Mar 28, 2026
commit ed4be31a0f016341f7d04f00a966e96f99765430
7 changes: 5 additions & 2 deletions bigframes/operations/ai.py
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
>>> print("START_OF_OUTPUT"); df1.ai.sim_join(df2, left_on='animal', right_on='animal', model=model, top_k=1) # doctest: +ELLIPSIS,+NORMALIZE_WHITESPACE
START_OF_OUTPUT
...
animal animal_1
...
0 monkey baboon
1 spider scorpion
<BLANKLINE>
Expand Down
6 changes: 4 additions & 2 deletions bigframes/session/__init__.py
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