def _get_classes()

in pytouch/datasets/digit.py [0:0]

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

    def _get_classes(self, root):
        classes = [d.name for d in os.scandir(root) if d.is_dir()]
        class_to_idx = {class_name: x for x, class_name in enumerate(classes)}
        return classes, class_to_idx