diff --git a/tests/trace/test_evaluate.py b/tests/trace/test_evaluate.py index 2a2ca4d407d2..22fd8485b12a 100644 --- a/tests/trace/test_evaluate.py +++ b/tests/trace/test_evaluate.py @@ -343,4 +343,4 @@ def score_simple(input, output): # However, the key part is that we have basically X + 2 splits, with the middle X # being equal. We want to ensure that the table_query is not called in sequence, # but rather lazily after each batch. - assert counts_split_by_table_query == [13, 700, 700, 700, 5] + assert counts_split_by_table_query == [13, 700, 700, 700, 5], log