-
Notifications
You must be signed in to change notification settings - Fork 294
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
[REVIEW] Fix MNMG failures in mg_dgl_extensions #2786
[REVIEW] Fix MNMG failures in mg_dgl_extensions #2786
Conversation
Codecov Report
Additional details and impacted files@@ Coverage Diff @@
## branch-22.12 #2786 +/- ##
===============================================
Coverage ? 60.45%
===============================================
Files ? 111
Lines ? 6514
Branches ? 0
===============================================
Hits ? 3938
Misses ? 2576
Partials ? 0 Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. ☔ View full report at Codecov. |
if isinstance(vertex_ids, int): | ||
vertex_ids = [vertex_ids] | ||
elif not isinstance(vertex_ids, | ||
(list, slice, self.__series_type)): | ||
vertex_ids = list(vertex_ids) |
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.
I think we should let the underlying library handle it rather than limiting location
to these types.
Also, this fails with GPU objects like cupy
, cudf.Series
, for numpy arrays we pay a cost.
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.
Thanks for adding the extra test coverage. Just had a few minor suggestions/fixes/typos
Co-authored-by: Rick Ratzel <3039903+rlratzel@users.noreply.github.com>
Co-authored-by: Rick Ratzel <3039903+rlratzel@users.noreply.github.com>
Thanks for the review Rick, Addressed all of your comments so should be ready for another round of reviews. |
@gpucibot merge |
This PR fixes the below errors that have popped up in MNMG testing.
(Work around [BUG] dask_cudf loc operation fails with cudf Series and cupy array fails cudf#11877)
vertex_col_names
tonode_col_names
to match DGL