typo

a9d4d469 · rusty1s · 26fa1b3e · a9d4d469
Commit a9d4d469 authored Oct 14, 2020 by rusty1s
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

torch_sparse/metis.py torch_sparse/metis.py +4 -3

No files found.
--- a/torch_sparse/metis.py
+++ b/torch_sparse/metis.py
@@ -19,9 +19,10 @@ def weight2metis(weight: torch.Tensor) -> Optional[torch.Tensor]:
    return weight_ratio.to(torch.long)
-def partition(src: SparseTensor, num_parts: int, recursive: bool = False,
+def partition(
-              weighted=False
+        src: SparseTensor, num_parts: int, recursive: bool = False,
-              ) -> Tuple[SparseTensor, torch.Tensor, torch.Tensor]:
+        weighted: bool = False
+) -> Tuple[SparseTensor, torch.Tensor, torch.Tensor]:
    assert num_parts >= 1
    if num_parts == 1: