Olá, Luiza! Tudo bem com você?
Somente com esse trecho de código não consegui identificar o que pode está ocasionando o problema encontrado.
Você conseguiria disponibilizar seu código? poderia colocar em um drive e compartilhar o link (lembrar de alterar as permissões para eu conseguir visualizar), ou poderia enviar o link do seu Notebook no Google Colab, caso esteja desenvolvendo por lá.
Você também poderia baixar o código da aula disponibilizado pelo instrutor nesse link e comparar com seu e analisar o que pode está diferente.
Fico no aguardo.