Skip to content

chore(bigframes): fix ai multimodal tests #16935

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
sycai merged 6 commits into from
May 5, 2026
+6 −6
Merged

chore(bigframes): fix ai multimodal tests #16935

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
07eca9b
fix(bigframes): fix ai multimodal tests
sycai May 5, 2026
dafb496
relax the test assertion
sycai May 5, 2026
9c2e992
relax assertions for all managed ai functions
sycai May 5, 2026
b5714b5
restore connection
sycai May 5, 2026
85b0298
fix lint
sycai May 5, 2026
dea795b
fix lint
sycai May 5, 2026
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
12 changes: 6 additions & 6 deletions packages/bigframes/tests/system/small/bigquery/test_ai.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -329,7 +329,7 @@ def test_ai_if(session):
max_error_ratio=0.5,
)

assert _contains_no_nulls(result)
assert len(result) == len(s1)
assert result.dtype == dtypes.BOOL_DTYPE


Expand All @@ -342,7 +342,7 @@ def test_ai_if_multi_model(session, bq_connection):

result = bbq.ai.if_((df["image"], " contains an animal"))

assert _contains_no_nulls(result)
assert len(result) == len(df)
assert result.dtype == dtypes.BOOL_DTYPE


Expand All @@ -351,7 +351,7 @@ def test_ai_classify(session):

result = bbq.ai.classify(s, ["animal", "plant"])

assert _contains_no_nulls(result)
assert len(result) == len(s)
assert result.dtype == dtypes.STRING_DTYPE


Expand All @@ -364,7 +364,7 @@ def test_ai_classify_multi_model(session, bq_connection):

result = bbq.ai.classify(df["image"], ["photo", "cartoon"])

assert _contains_no_nulls(result)
assert len(result) == len(df)
assert result.dtype == dtypes.STRING_DTYPE


Expand All @@ -374,7 +374,7 @@ def test_ai_score(session):

result = bbq.ai.score(prompt)

assert _contains_no_nulls(result)
assert len(result) == len(s)
assert result.dtype == dtypes.FLOAT_DTYPE


Expand All @@ -386,7 +386,7 @@ def test_ai_score_multi_model(session):

result = bbq.ai.score(prompt)

assert _contains_no_nulls(result)
assert len(result) == len(df)
assert result.dtype == dtypes.FLOAT_DTYPE


Expand Down
Loading