529c4bc269f6c3a265d49b97393cbc6ecbcf5f94,python/hypertools/plot/plot.py,,plot,#Any#,26

Before Change

        categories = list(set(np.sort(point_colors)))
        x_stacked = np.vstack(x)
        x_reshaped = [[] for i in categories]
        for idx,point in enumerate(point_colors):
            x_reshaped[categories.index(point)].append(x_stacked[idx])
        x = [np.vstack(i) for i in x_reshaped]

    if "style" in kwargs:
        sns.set(style=kwargs["style"])

After Change

    ////HYPERTOOLS-SPECIFIC ARG PARSING////

    if "n_clusters" in kwargs:
        n_clusters=kwargs["n_clusters"]

        if "ndims" in kwargs:
            ndims = kwargs["ndims"]
        else:
            ndims = 3

        cluster_labels = get_clusters(x, ndims, n_clusters)
        x = reshape_data(x,cluster_labels)
        del kwargs["n_clusters"]
        
        if "point_colors" in kwargs:
            warnings.warn("n_clusters overrides point_colors, ignoring point_colors.")
            del kwargs["point_colors"]

    // handle point_colors flag
    if "point_colors" in kwargs:
        point_colors=kwargs["point_colors"]
        del kwargs["point_colors"]

Instances