def count_elements_in_nested_sequence()

in augmentation/utils.py [0:0]


def count_elements_in_nested_sequence(lst: List, sequence_type = List):
    if not isinstance(lst, sequence_type):
        return 1
    return sum(count_elements_in_nested_sequence(e, sequence_type) for e in lst)