9685df29213e5943d2774f80e21564f4204823ec,secuml/core/data/annotations.py,Annotations,get_annotated_ids,#Annotations#Any#Any#,155

Before Change

        if family is not None:
            return [i for i in annotated_ids if self.get_family(i) == family]
        else:
            return annotated_ids

    def get_unlabeled_ids(self):
        return [i for i in self.ids.get_ids() if not self.is_annotated(i)]

After Change

        else:
            mask = self.labels == label_str_to_bool(label)
        if family is not None:
            family_mask = self.families == family
            mask = np.logical_and(mask, family_mask)
        return self.ids.ids[mask]

    def get_unlabeled_ids(self):
        return self.ids.ids[self.labels == None]  // NOQA: 711

Instances