Skip to content
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

Add Additional Checks to get_edgelist and get_dask_edgelist #4256

Merged
merged 14 commits into from
May 2, 2024

Conversation

nv-rliu
Copy link
Contributor

@nv-rliu nv-rliu commented Mar 18, 2024

Closes #4241

This PR adds an additional check to the get_edgelist() and get_dask_edgelist() functions in the Datasets API.

This ensures that, when retrieving an edge-list, the internal (self._edgelist) type is verified to ensure that the object is SG or MG.

In addition, minor improvements have also been made utils/test_dataset.py to be more thorough with type checks.

@nv-rliu nv-rliu added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change datasets labels Mar 18, 2024
@nv-rliu nv-rliu requested review from a team as code owners March 18, 2024 20:32
@nv-rliu nv-rliu changed the base branch from branch-24.04 to branch-24.06 March 18, 2024 20:32
@nv-rliu nv-rliu removed request for a team March 18, 2024 20:32
@nv-rliu nv-rliu added this to the 24.06 milestone Mar 18, 2024
@github-actions github-actions bot added the ci label Mar 26, 2024
@nv-rliu nv-rliu removed request for a team March 26, 2024 16:26
@github-actions github-actions bot removed the ci label Mar 26, 2024
@nv-rliu nv-rliu requested a review from rlratzel April 4, 2024 15:43
@rlratzel
Copy link
Contributor

@nv-rliu and I discussed this offline earlier today and I believe he has a change in mind for this PR that eliminates the need for the additional overwrite arg. I'm going to hold off on a review until that change is in place.

@nv-rliu nv-rliu changed the title Add overwrite Argument to datasets.get_graph() Add Additional Checks to get_edgelist and get_dask_edgelist Apr 10, 2024
@nv-rliu nv-rliu requested a review from rlratzel April 10, 2024 16:40
Copy link
Contributor

@rlratzel rlratzel left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks, Ralph!

@rlratzel
Copy link
Contributor

rlratzel commented May 2, 2024

/merge

@rapids-bot rapids-bot bot merged commit 581bd9c into rapidsai:branch-24.06 May 2, 2024
131 checks passed
@nv-rliu nv-rliu deleted the datasets-auto-unload branch May 17, 2024 18:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
improvement Improvement / enhancement to an existing function non-breaking Non-breaking change python
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[FEA]: Add parameter to prevent persisted edgelists in datasets API
2 participants