Skip to content

Commit 6b84dfc

Browse files
committed
revert no_null check
1 parent cd87e73 commit 6b84dfc

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

  • packages/bigframes/tests/system/small/bigquery

packages/bigframes/tests/system/small/bigquery/test_ai.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -342,7 +342,7 @@ def test_ai_if_multi_model(session, bq_connection):
342342

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

345-
assert len(result) == len(df) # nulls are allowed
345+
assert _contains_no_nulls(result)
346346
assert result.dtype == dtypes.BOOL_DTYPE
347347

348348

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

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

367-
assert len(result) == len(df) # nulls are allowed
367+
assert _contains_no_nulls(result)
368368
assert result.dtype == dtypes.STRING_DTYPE
369369

370370

0 commit comments

Comments
 (0)