Skip to content

Commit 0a70173

Browse files
committed
fix doctest
1 parent 7cb3d55 commit 0a70173

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

bigframes/bigquery/_operations/search.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -164,12 +164,12 @@ def vector_search(
164164
... query=search_query,
165165
... distance_type="cosine",
166166
... query_column_to_search="another_embedding",
167-
... top_k=2)
167+
... top_k=2).sort_values("id")
168168
query_id embedding another_embedding id my_embedding distance
169-
1 cat [3. 5.2] [3.3 5.2] 2 [2. 4.] 0.005181
170-
0 dog [1. 2.] [0.7 2.2] 4 [1. 3.2] 0.000013
171169
1 cat [3. 5.2] [3.3 5.2] 1 [1. 2.] 0.005181
170+
1 cat [3. 5.2] [3.3 5.2] 2 [2. 4.] 0.005181
172171
0 dog [1. 2.] [0.7 2.2] 3 [1.5 7. ] 0.004697
172+
0 dog [1. 2.] [0.7 2.2] 4 [1. 3.2] 0.000013
173173
<BLANKLINE>
174174
[4 rows x 6 columns]
175175

0 commit comments

Comments
 (0)