code clean

torch_sparse/metis.py

@@ -18,9 +18,8 @@ def metis_wgt(x):
     return (x_ratio * arange + tick).long(), tick, arange
 
 
-def partition(
-        src: SparseTensor, num_parts: int, recursive: bool = False
-) -> Tuple[SparseTensor, torch.Tensor, torch.Tensor]:
+def partition(src: SparseTensor, num_parts: int, recursive: bool = False
+              ) -> Tuple[SparseTensor, torch.Tensor, torch.Tensor]:
     rowptr, col = src.storage.rowptr().cpu(), src.storage.col().cpu()
     edge_wgt = src.storage.value().cpu()
     edge_wgt = metis_wgt(edge_wgt)[0]