def _find_classes()

in src/sagemaker_defect_detection/dataset/neu.py [0:0]

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

    def _find_classes(self):
        classes = sorted([d.name for d in os.scandir(str(self.root)) if d.is_dir()])
        class_to_idx = {cls_name: i for i, cls_name in enumerate(classes, 1)}  # no bg label in NEU
        return classes, class_to_idx