Altered the transform function to use the new bipartite option in compute_membership_strength.  The old code had the bug that if the ith new_point was nearest neighbours of the ith old point their membership strength would be set to 0.
@@ -2681,7 +2681,7 @@ class UMAP(BaseEstimator):
)
rows, cols, vals, dists = compute_membership_strengths(
- indices, dists, sigmas, rhos
+ indices, dists, sigmas, rhos, bipartite=True
)
graph = scipy.sparse.coo_matrix(
