Skip to content
This repository was archived by the owner on Apr 1, 2026. It is now read-only.

fix: deflake ai_gen_bool multimodel test #2085

Merged
sycai merged 6 commits into from
Sep 16, 2025
Merged

fix: deflake ai_gen_bool multimodel test #2085

Changes from 2 commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
9c94744
fix: deflake ai_gen_bool multimodel test
sycai Sep 15, 2025
c98c9e7
fix lint
sycai Sep 15, 2025
97fa173
fix doctest too
sycai Sep 15, 2025
1ce7a32
consolidates tests under system/small
sycai Sep 15, 2025
bf7d249
Merge branch 'main' into sycai_ai_gen_bool_mm_fix
sycai Sep 15, 2025
c9fec4c
fix doctest
sycai Sep 15, 2025
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
11 changes: 2 additions & 9 deletions tests/system/large/bigquery/test_ai.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,9 +12,6 @@
# See the License for the specific language governing permissions and
# limitations under the License.

import pandas as pd
import pandas.testing

import bigframes.bigquery as bbq


Expand All @@ -27,9 +24,5 @@ def test_ai_generate_bool_multi_model(session):
"result"
)

pandas.testing.assert_series_equal(
result.to_pandas(),
pd.Series([True, True, False, False, False], name="result"),
check_dtype=False,
check_index=False,
)
# Make sure all elements are not null
assert len(result) == result.count()
Comment thread
sycai marked this conversation as resolved.
Outdated