O código em Python funciona em qualquer IDE, porém, podem haver diferenças de acordo com a biblioteca que está instalada no seu ambiente e precisaria fazer testes. O ideal é executar o código da aula no Google Colab para não perder tempo se preocupando com bibliotecas, já que nós procuramos manter o arquivo atualizado e com correções no caso de alterações em bibliotecas

Sugiro primeiro rodar no Colab e depois que estiver funcionando mudar para outro ambiente