Olá Fábio,
Sim, é o mesmo código da aula.
Inclusive peguei o mesmo que está naquele arquivo do Colab que foi anexado no conteúdo pra ter certeza de que estava ok.
Se não estiver enganado, parece que há algum problema com o nome das colunas serem do tipo ‘str’ e ‘int’.
No Colab quando executo ele roda mas alerta sobre isso em uma versão futura. Já no VS Code não consigo executar.