def get_class_labels()

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

• 5 lines of code
• 3 McCabe index (conditional complexity)

    def get_class_labels(self) -> List[str]:
        """
        Return the model's classification labels as a sorted list.
        """
        labels = [self._config.id2label[x] for x in range(self._config.num_labels)]

        if len(labels) != len(set(labels)):
            labels = self._rename_duplicate_labels(labels)

        return labels