def _rename_duplicate_labels()

in src/exporters/coreml/config.py [0:0]

• 11 lines of code
• 4 McCabe index (conditional complexity)

    def _rename_duplicate_labels(self, labels):
        """
        Renames duplicate label names. Core ML puts the labels as keys into a dictionary,
        and so all the label names need to be unique.
        """
        unique_labels = []
        used_labels = set()

        for label in labels:
            while label in used_labels:
                label = label + "_duplicate"

            used_labels.add(label)
            unique_labels.append(label)

        if len(unique_labels) != len(set(unique_labels)):
            raise AssertionError("Unable to remove duplicates from the provided labels")

        return unique_labels