diff --git a/bigframes/operations/ai.py b/bigframes/operations/ai.py index 6516620b536..456ef3ecda3 100644 --- a/bigframes/operations/ai.py +++ b/bigframes/operations/ai.py @@ -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 diff --git a/bigframes/session/__init__.py b/bigframes/session/__init__.py index 710b3701fae..75be3022d7d 100644 --- a/bigframes/session/__init__.py +++ b/bigframes/session/__init__.py @@ -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 diff --git a/third_party/bigframes_vendored/pandas/io/gbq.py b/third_party/bigframes_vendored/pandas/io/gbq.py index 3190c92b921..79faa53de2b 100644 --- a/third_party/bigframes_vendored/pandas/io/gbq.py +++ b/third_party/bigframes_vendored/pandas/io/gbq.py @@ -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