add type hints to group_distance

Author: mikedh

trimesh/grouping.py

@@ -9,7 +9,7 @@
 
 from . import util
 from .constants import log, tol
-from .typed import ArrayLike, Integer, NDArray, Optional
+from .typed import ArrayLike, Integer, NDArray, Number, Optional, Sequence, Tuple
 
 try:
     from scipy.spatial import cKDTree
@@ -666,14 +666,16 @@ def group_vectors(vectors, angle=1e-4, include_negative=False):
     return new_vectors, groups
 
 
-def group_distance(values, distance):
+def group_distance(
+    values: ArrayLike, distance: Number
+) -> Tuple[NDArray[np.float64], Sequence]:
     """
     Find non-overlapping groups of points where no two points in a
     group are farther than 2*distance apart.
 
     Parameters
     ---------
-    values :   (n, d) float
+    values : (n, d) float
         Points of dimension d
     distance : float
         Max distance between points in a cluster