Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Oct 18, 2024
1 parent 1e2bd6f commit fb432d8
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions examples/distributed/wholegraph/graph_store.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,7 @@ class WholeGraphEdgeAttr(EdgeAttr):
r"""Edge attribute class for WholeGraph GraphStore enforcing layout to be CSC."""
def __init__(
self,
edge_type: Optional[
EdgeType] = None,
edge_type: Optional[EdgeType] = None,
is_sorted: bool = False,
size: Optional[Tuple[int, int]] = None,
):
Expand Down Expand Up @@ -85,8 +84,7 @@ def __init__(self, pyg_data, format='wholegraph'):
if 'adj_t' not in edge_store:
row, col = None, None
if dist_shmem.get_local_rank() == 0:
row, col, _ = edge_store.csc(
)
row, col, _ = edge_store.csc()
row = dist_shmem.to_shmem(row)
col = dist_shmem.to_shmem(col)
size = edge_store.size()
Expand Down

0 comments on commit fb432d8

Please sign in to comment.