fix flake error

rusty1s · Mar 1, 2021 · 9b0759b · 9b0759b
1 parent 90c0c4b
commit 9b0759b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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