Skip to content

GitLab

Commit 3dead4d1 authored by rusty1s's avatar rusty1s
Browse files

typo

parent a29aabed
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torch_sparse/metis.py
View file @ 3dead4d1
...@@ -21,7 +21,7 @@ def weight2metis(weight: torch.Tensor) -> Optional[torch.Tensor]: ...@@ -21,7 +21,7 @@ def weight2metis(weight: torch.Tensor) -> Optional[torch.Tensor]:
def partition( def partition(
src: SparseTensor, num_parts: int, recursive: bool = False, src: SparseTensor, num_parts: int, recursive: bool = False,
weighted: bool = False, node_weight: torch.tensor = None weighted: bool = False, node_weight: torch.Tensor = None
) -> Tuple[SparseTensor, torch.Tensor, torch.Tensor]: ) -> Tuple[SparseTensor, torch.Tensor, torch.Tensor]:
assert num_parts >= 1 assert num_parts >= 1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment