Здравствуйте! Начал изучать нейронные сети и решил повторить все по данной  
 статье
Установил все сопутствующие библиотеки и остановился на данном шаге 

Так как возникает следующая ошибка 

Версия tensorflow 2.4.1  
Что я сделал не правильно и как избавиться от ошибки? 
Если можно дайте ответ максимально просто и понятно так как я новичок )