-
Notifications
You must be signed in to change notification settings - Fork 532
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Adding tol to dask test_kmeans #3426
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
looks good (pending copyright year update), just had one question before we merge
@@ -193,4 +193,4 @@ def test_score(nrows, ncols, nclusters, n_parts, | |||
local_model = cumlModel.get_combined_model() | |||
expected_score = local_model.score(X_train.compute()) | |||
|
|||
assert actual_score == expected_score | |||
assert abs(actual_score - expected_score) < 1e-3 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why was 1e-3 in particular chosen?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Because it's small enough compared to the numbers in the assertion (15k and 5k).
The error reported in the issue were in the order of 2e-4.
@gpucibot merge |
Closes #2969.
This PR add tolerance to dask test_kmeans in order to be flexible with small accumulation mistakes in a distributed context.