Olá! Sempre atualizamos os scripts quando erros são reportados
Acabamos de ver que como o colab foi atualizado com versões mais recentes do Python, não é possível mais rodar por ele. Para funcionar, teria que instalar o ambiente diretamente na máquina local com as seguintes versões:
python==3.7
tensorflow==2.3.0
setuptools<58
Colocamos esse código na lista de erratas para postarmos uma atualização com bibliotecas mais novas