ЯЗЫК PYTHON
Напишите программу, которая определяет, верно ли, что введённое число состоит из одинаковых цифр (как, например, число 222).

Входные данные
Входная строка содержит одно натуральное число.

Выходные данные
Программа должна вывести слово 'YES', если все цифры в числе одинаковые, и слово 'NO', если это неверно.

программа не работает

бываетвот одно дело, когда программа ничего не возвращает, это один вариант "не работает", а вот когда появляются какие-то ошибки, то другойу вас какой вариант?

первый

а какие были ваши действия?

def digits():

   num = input()

   if len(set(num)) == 1:

       print('YES')

   else:

       print('NO')