import pathlib


def list_dirs(directory):
    """Returns all directories in a train directory
    """
    return [folder for folder in pathlib.Path(directory).iterdir()
            if folder.is_dir()]