renamed partial function to match non-small_data name
@@ -1175,10 +1175,10 @@ def pairwise_special_metric(X, Y=None, metric="hellinger", kwds=None):
kwd_vals = tuple(kwds.values())
@numba.njit(fastmath=True)
- def _pairwise_numba(_X, _Y=None):
+ def _partial_metric(_X, _Y=None):
return metric(_X, _Y, *kwd_vals)
- return pairwise_distances(X, Y, metric=_pairwise_numba)
+ return pairwise_distances(X, Y, metric=_partial_metric)
else:
special_metric_func = named_distances[metric]
return parallel_special_metric(X, Y, metric=special_metric_func)
