Você está vendo a versão anterior da nova experiência da Alura que estamos preparando para você. Em breve, ela ganha uma identidade visual novinha totalmente pensada em potencializar seus estudos!
Olá, Henrique, tudo bem?
Analisei o seu código e ele está certinho! Não deveria haver nenhum motivo para esse erro específico nesse trecho de código. O que pode ter ocorrido é algum erro residual que ocasionou esse comportamento.
Para resolver esse problema, peço que reinicie o ambiente de execução no Colab.
Em seguida, tente executar o código novamente.
Espero que ajude a solucionar o problema.
Caso surja alguma dúvida, fico à disposição.