task4.py

import os


def get_next(class_name: str) -> str:
    """
    Возвращает относительный путь для объекта класса

    Данная функция возвращает относительрный путь для объекта класса, переданного
    в функцию
    Parameters
    ----------
    class_name : str
      Имя класса
    Returns
    -------
    str
    Относительный путь к объекту
    """
    path = os.path.join('dataset', class_name)
    class_names = os.listdir(path)
    class_names.append(None)
    for i in range(len(class_names)):
        if class_names[i] is not None:
            yield os.path.join(path, class_names[i])
        elif class_names[i] is None:
            yield None   

print(*get_next('polarbear'))