-
Notifications
You must be signed in to change notification settings - Fork 554
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
Use estimator tags to improve sparse error handling #6151
Use estimator tags to improve sparse error handling #6151
Conversation
Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually. Contributors can view more details about this message here. |
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.
One stray print statement and one question on testing.
@pytest.mark.parametrize("estimator_name", list(estimators.keys())) | ||
def test_sparse_support(estimator_name): | ||
X_sparse = csr_matrix([[0, 1], [1, 0]]) | ||
print(X_sparse.shape[0]) |
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.
print(X_sparse.shape[0]) |
|
||
|
||
@pytest.mark.parametrize("estimator_name", list(estimators.keys())) | ||
def test_sparse_support(estimator_name): |
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.
How would we feel about adding a test that confirms that an estimator without sparse support raises the correct exception on sparse input?
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.
Added them in a separate file to test vanilla cuML exceptions
…h-sparse-handling
…s, and remove them from exception test
@@ -122,13 +122,19 @@ def custom_weights(distances): | |||
assert acc > 0.7, "Accuracy should be reasonable with custom weights" | |||
|
|||
|
|||
@pytest.mark.xfail( | |||
reason="cuML and sklearn don't have matching exceptions yet" |
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.
Do we have an issue logged for this?
/merge |
No description provided.