We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a120bae commit bd23ec3Copy full SHA for bd23ec3
tests/system/large/ml/test_linear_model.py
@@ -140,7 +140,8 @@ def test_unordered_mode_linear_regression_configure_fit_score_predict(
140
start_execution_count = end_execution_count
141
result = model.score(X_train, y_train).to_pandas()
142
end_execution_count = df._block._expr.session._metrics.execution_count
143
- assert end_execution_count - start_execution_count == 1
+ # The score function and to_pandas each initiate one query.
144
+ assert end_execution_count - start_execution_count == 2
145
146
utils.check_pandas_df_schema_and_index(
147
result, columns=utils.ML_REGRESSION_METRICS, index=1
0 commit comments