Fix RSL for #597

Author: lmcinnes

hdbscan/robust_single_linkage_.py

@@ -23,8 +23,22 @@
 # Author: Leland McInnes <[email protected]>
 #
 # License: BSD 3 clause
-
-FAST_METRICS = KDTree.valid_metrics + BallTree.valid_metrics
+KDTREE_VALID_METRICS = ["euclidean", "l2", "minkowski", "p", "manhattan", "cityblock", "l1", "chebyshev", "infinity"]
+BALLTREE_VALID_METRICS = KDTREE_VALID_METRICS + [
+    "braycurtis",
+    "canberra",
+    "dice",
+    "hamming",
+    "haversine",
+    "jaccard",
+    "mahalanobis",
+    "rogerstanimoto",
+    "russellrao",
+    "seuclidean",
+    "sokalmichener",
+    "sokalsneath",
+]
+FAST_METRICS = KDTREE_VALID_METRICS + BALLTREE_VALID_METRICS
 
 
 def _rsl_generic(X, k=5, alpha=1.4142135623730951, metric='euclidean',
@@ -266,7 +280,7 @@ def robust_single_linkage(X, cut, k=5, alpha=1.4142135623730951,
             # We can't do much with sparse matrices ...
             single_linkage_tree = memory.cache(_rsl_generic)(
                 X, k, alpha, metric, **kwargs)
-        elif metric in KDTree.valid_metrics:
+        elif metric in KDTREE_VALID_METRICS:
             # Need heuristic to decide when to go to boruvka;
             # still debugging for now
             if X.shape[1] > 128: