Skip to content

Commit 97fa173

Browse files
committed
fix doctest too
1 parent c98c9e7 commit 97fa173

File tree

1 file changed

+3
-3
lines changed
  • bigframes/bigquery/_operations

1 file changed

+3
-3
lines changed

bigframes/bigquery/_operations/ai.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,13 +47,13 @@ def generate_bool(
4747
... "col_1": ["apple", "bear", "pear"],
4848
... "col_2": ["fruit", "animal", "animal"]
4949
... })
50-
>>> bbq.ai_generate_bool((df["col_1"], " is a ", df["col_2"]))
50+
>>> bbq.ai.generate_bool((df["col_1"], " is a ", df["col_2"]))
5151
0 {'result': True, 'full_response': '{"candidate...
5252
1 {'result': True, 'full_response': '{"candidate...
5353
2 {'result': False, 'full_response': '{"candidat...
5454
dtype: struct<result: bool, full_response: string, status: string>[pyarrow]
5555
56-
>>> bbq.ai_generate_bool((df["col_1"], " is a ", df["col_2"])).struct.field("result")
56+
>>> bbq.ai.generate_bool((df["col_1"], " is a ", df["col_2"])).struct.field("result")
5757
0 True
5858
1 True
5959
2 False
@@ -66,7 +66,7 @@ def generate_bool(
6666
... }
6767
... }
6868
... }
69-
>>> bbq.ai_generate_bool(
69+
>>> bbq.ai.generate_bool(
7070
... (df["col_1"], " is a ", df["col_2"]),
7171
... endpoint="gemini-2.5-pro",
7272
... model_params=model_params,

0 commit comments

Comments
 (0)