Merge branch 'main' into shuowei-job-history

Author: shuoweil

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
             <BLANKLINE>

bigframes/session/__init__.py

@@ -701,9 +701,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>

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
             <BLANKLINE>