def METHOD_NAME(iris, iris_selection):
    return UMAP(
        n_neighbors=10,
        min_dist=0.01,
        random_state=42,
        force_approximation_algorithm=True,
    ).fit(iris.data[iris_selection])