diff --git a/python/cuml/cuml/manifold/umap.pyx b/python/cuml/cuml/manifold/umap.pyx index 9e4f3a6551..83f29f76c8 100644 --- a/python/cuml/cuml/manifold/umap.pyx +++ b/python/cuml/cuml/manifold/umap.pyx @@ -341,7 +341,7 @@ class UMAP(UniversalBase, "sokalmichener": "NotImplemented", "yule": "NotImplemented", "ll_dirichlet": "NotImplemented", - "russelrao": "NotImplemented", + "russellrao": "NotImplemented", "kulsinski": "NotImplemented", "dice": "NotImplemented", "wminkowski": "NotImplemented", diff --git a/python/cuml/cuml/tests/experimental/accel/estimators_hyperparams/test_accel_umap.py b/python/cuml/cuml/tests/experimental/accel/estimators_hyperparams/test_accel_umap.py index 221082d32a..67ace7c3fd 100644 --- a/python/cuml/cuml/tests/experimental/accel/estimators_hyperparams/test_accel_umap.py +++ b/python/cuml/cuml/tests/experimental/accel/estimators_hyperparams/test_accel_umap.py @@ -59,7 +59,7 @@ def test_umap_min_dist(manifold_data, min_dist): "sokalmichener", "yule", "ll_dirichlet", - "russelrao", + "russellrao", "kulsinski", "dice", "wminkowski", @@ -72,6 +72,7 @@ def test_umap_metric(manifold_data, metric): # haversine only works for 2D data if metric == "haversine": X = X[:, :2] + umap = UMAP(metric=metric, random_state=42) X_embedded = umap.fit_transform(X) trust = trustworthiness(X, X_embedded, n_neighbors=5)