Skip to content

Commit bd23ec3

Browse files
authored
chore: score query count assertion fix. (#1093)
1 parent a120bae commit bd23ec3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/system/large/ml/test_linear_model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,8 @@ def test_unordered_mode_linear_regression_configure_fit_score_predict(
140140
start_execution_count = end_execution_count
141141
result = model.score(X_train, y_train).to_pandas()
142142
end_execution_count = df._block._expr.session._metrics.execution_count
143-
assert end_execution_count - start_execution_count == 1
143+
# The score function and to_pandas each initiate one query.
144+
assert end_execution_count - start_execution_count == 2
144145

145146
utils.check_pandas_df_schema_and_index(
146147
result, columns=utils.ML_REGRESSION_METRICS, index=1

0 commit comments

Comments
 (0)