diff --git a/umap/distances.py b/umap/distances.py index 3701efa3..6027409b 100644 --- a/umap/distances.py +++ b/umap/distances.py @@ -1305,4 +1305,4 @@ def _partial_metric(_X, _Y=None): return pairwise_distances(X, Y, metric=_partial_metric) else: special_metric_func = named_distances[metric] - return chunked_parallel_special_metric(X, Y, metric=special_metric_func) + return parallel_special_metric(X, Y, metric=special_metric_func)