21a6d7c02d8919ac29b5aa0af8992ba13f0a3ddb,prody/chromatin/cluster.py,,calcGNMDomains,#Any#Any#,190

Before Change

    labels_ = method(V, **kwargs)

    if np.all(mask):
        labels = labels_
    else:
        labels = np.empty(len(mask))
        labels.fill(np.nan)
        labels[mask] = labels_

        currlbl = labels_[np.argmax(~np.isnan(labels_))]

        for i, l in enumerate(labels):
            if np.isnan(l):
                labels[i] = currlbl
            elif currlbl != l:
                currlbl = l

    return labels

After Change

    labels_ = method(V, **kwargs)

    if hasattr(modes, "_model"):
        model = modes._model
        if isinstance(model, MaskedGNM):
            labels = model._extend(labels_, np.nan)
            currlbl = labels_[0]

            for i, l in enumerate(labels):
                if np.isnan(l):
                    labels[i] = currlbl
                elif currlbl != l:
                    currlbl = l

    return labels
    

Instances