Указать тип данных в функции python