Skip to content

Commit

Permalink
Merge branch 'fix-loader-bugs' of https://github.com/alexbarghi-nv/cu…
Browse files Browse the repository at this point in the history
…graph into fix-loader-bugs
  • Loading branch information
alexbarghi-nv committed Nov 8, 2023
2 parents 0f4693c + e207f98 commit 65f50a3
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py
Original file line number Diff line number Diff line change
Expand Up @@ -214,7 +214,7 @@ def __init__(
num_nodes_dict: Dict[str, int],
*,
multi_gpu: bool = False,
order: str = "CSC",
order: str = "CSR",
):
"""
Constructs a new CuGraphStore from the provided
Expand Down Expand Up @@ -260,11 +260,11 @@ def __init__(
Whether the store should be backed by a multi-GPU graph.
Requires dask to have been set up.
order: str (Optional ["CSR", "CSC"], default = CSC)
The order to use for sampling. Should nearly always be CSC
unless there is a specific expectation of "reverse" sampling.
It is also not uncommon to use CSR order for correctness
testing, which some cuGraph-PyG tests do.
order: str (Optional ["CSR", "CSC"], default = CSR)
The order to use for sampling. CSR corresponds to the
standard OGB dataset order that is usually used in PyG.
CSC order constructs the same graph as CSR, but with
edges in the opposite direction.
"""

if None in G:
Expand Down

0 comments on commit 65f50a3

Please sign in to comment.