We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent cd87e73 commit 6b84dfcCopy full SHA for 6b84dfc
1 file changed
packages/bigframes/tests/system/small/bigquery/test_ai.py
@@ -342,7 +342,7 @@ def test_ai_if_multi_model(session, bq_connection):
342
343
result = bbq.ai.if_((df["image"], " contains an animal"))
344
345
- assert len(result) == len(df) # nulls are allowed
+ assert _contains_no_nulls(result)
346
assert result.dtype == dtypes.BOOL_DTYPE
347
348
@@ -364,7 +364,7 @@ def test_ai_classify_multi_model(session, bq_connection):
364
365
result = bbq.ai.classify(df["image"], ["photo", "cartoon"])
366
367
368
assert result.dtype == dtypes.STRING_DTYPE
369
370
0 commit comments