typo2

torch_sparse/metis.py

@@ -21,7 +21,7 @@ def weight2metis(weight: torch.Tensor) -> Optional[torch.Tensor]:
 
 def partition(
     src: SparseTensor, num_parts: int, recursive: bool = False,
-    weighted: bool = False, node_weight: torch.Tensor = None
+    weighted: bool = False, node_weight: Optional[torch.Tensor] = None
 ) -> Tuple[SparseTensor, torch.Tensor, torch.Tensor]:
 
     assert num_parts >= 1