diff --git a/torch_cluster/fps.py b/torch_cluster/fps.py index 450c565b..2ca54d23 100644 --- a/torch_cluster/fps.py +++ b/torch_cluster/fps.py @@ -5,13 +5,13 @@ @torch.jit._overload # noqa -def fps(src, batch=None, ratio=None, random_start=True): +def fps(src, batch=None, ratio=None, random_start=True): # noqa # type: (Tensor, Optional[Tensor], Optional[float], bool) -> Tensor pass @torch.jit._overload # noqa -def fps(src, batch=None, ratio=None, random_start=True): +def fps(src, batch=None, ratio=None, random_start=True): # noqa # type: (Tensor, Optional[Tensor], Optional[Tensor], bool) -> Tensor pass