Oi Eduardo, tudo bem?
Neste caso, sugiro reiniciar a sessão do Colab para que possíveis resquícios de códigos não afete a saída. Siga os passos abaixo:
No menu de ferramentas do Google Colaboratory, clicar em "Ambientes de execução";
Selecionar a opção "Reiniciar sessão e executar tudo".
Para realizar essas etapas, você pode acompanhar a imagem abaixo:
Se o problema persistir, você poderá compartilhar print da execução do código com a saída. Assim, poderei te ajudar mais assertivamente.
Espero ter ajudado.
Abraços e bons estudos!
Caso este post tenha lhe ajudado, por favor, marcar como solucionado ✓. Bons Estudos!