A minha rodou também, era só incluir o tipo de variável:
def habilitacao():
idade = input("Qual a sua idade? ")
idade = int (idade)
if idade >= 18:
print('Pode tirar a habilitação')
else:
tempo = 18 - idade
print(f"Calma... espere {tempo} ano(s) para tirar habilitação")
habilitacao()