Help me with W1A3, There is something wrong with the given code!

Here is the complete list. Basically, in Coursera environment, TF version is mostly 2.3. Some assignments need 2.4. Of course, it does not mean an assignment does not run on like TF2.9. But, all are at your own risk.

Package Version


absl-py 0.9.0
alembic 1.4.2
astor 0.8.1
astunparse 1.6.3
async-generator 1.10
attrs 19.3.0
backcall 0.1.0
beautifulsoup4 4.9.0
bleach 3.1.4
blinker 1.4
bokeh 2.0.1
Bottleneck 1.3.2
brotlipy 0.7.0
cachetools 4.1.0
certifi 2020.4.5.1
certipy 0.1.3
cffi 1.14.0
chardet 4.0.0
click 7.1.2
cloudpickle 1.4.1
conda 4.8.2
conda-package-handling 1.6.0
cryptography 2.9.2
cycler 0.10.0
Cython 0.29.17
cytoolz 0.10.1
dask 2.15.0
decorator 4.4.2
defusedxml 0.6.0
dill 0.3.1.1
distributed 2.15.2
emoji 1.2.0
entrypoints 0.3
fastcache 1.1.0
ffmpeg 1.4
fsspec 0.7.3
gast 0.3.3
gmpy2 2.1.0b1
google-auth 1.14.1
google-auth-oauthlib 0.4.1
google-pasta 0.2.0
grpcio 1.28.1
h5py 2.10.0
HeapDict 1.0.1
idna 2.9
imageio 2.8.0
importlib-metadata 1.6.0
ipykernel 5.2.1
ipympl 0.5.6
ipython 7.14.0
ipython-genutils 0.2.0
ipywidgets 7.5.1
jedi 0.17.0
Jinja2 2.11.2
joblib 1.0.0
json5 0.9.0
jsonschema 3.2.0
jupyter-client 6.1.3
jupyter-core 4.6.3
jupyter-telemetry 0.0.5
jupyterhub 1.1.0
jupyterlab 2.1.1
jupyterlab-server 1.1.1
Keras-Applications 1.0.8
Keras-Preprocessing 1.1.2
kiwisolver 1.2.0
llvmlite 0.31.0
locket 0.2.0
Mako 1.1.0
Markdown 3.2.1
MarkupSafe 1.1.1
matplotlib 3.2.1
mido 1.2.9
mistune 0.8.4
mock 4.0.2
more-itertools 8.6.0
mpmath 1.1.0
msgpack 1.0.0
music21 6.5.0
nbconvert 5.6.1
nbformat 5.0.6
networkx 2.4
notebook 6.0.3
numba 0.48.0
numexpr 2.7.1
numpy 1.18.4
oauthlib 3.0.1
olefile 0.46
opt-einsum 3.2.1
packaging 20.1
pamela 1.0.0
pandas 1.0.3
pandocfilters 1.4.2
parso 0.7.0
partd 1.1.0
patsy 0.5.1
pexpect 4.8.0
pickleshare 0.7.5
Pillow 7.1.2
pip 20.1
prometheus-client 0.7.1
prompt-toolkit 3.0.5
protobuf 3.11.4
psutil 5.7.0
ptyprocess 0.6.0
pyasn1 0.4.8
pyasn1-modules 0.2.8
pycosat 0.6.3
pycparser 2.20
pycurl 7.43.0.5
pydub 0.24.0
Pygments 2.6.1
PyJWT 1.7.1
pyOpenSSL 19.1.0
pyparsing 2.4.7
pyrsistent 0.16.0
PySocks 1.7.1
python-dateutil 2.8.1
python-editor 1.0.4
python-json-logger 0.1.11
pytz 2020.1
PyWavelets 1.1.1
PyYAML 5.3.1
pyzmq 19.0.0
requests 2.23.0
requests-oauthlib 1.3.0
rsa 4.0
ruamel-yaml 0.15.80
ruamel.yaml 0.16.6
ruamel.yaml.clib 0.2.0
scikit-image 0.16.2
scikit-learn 0.22.2.post1
scipy 1.4.1
seaborn 0.10.1
Send2Trash 1.5.0
setuptools 46.1.3.post20200325
six 1.14.0
sortedcontainers 2.1.0
soupsieve 1.9.4
SQLAlchemy 1.3.16
statsmodels 0.11.1
sympy 1.5.1
tables 3.6.1
tblib 1.6.0
tensorboard 2.4.1
tensorboard-plugin-wit 1.8.0
tensorflow 2.3.0
tensorflow-estimator 2.3.0
termcolor 1.1.0
terminado 0.8.3
testpath 0.4.4
toolz 0.10.0
tornado 6.0.4
tqdm 4.45.0
traitlets 4.3.3
typing-extensions 3.7.4.2
urllib3 1.25.9
vincent 0.4.4
wcwidth 0.1.9
webcolors 1.11.1
webencodings 0.5.1
Werkzeug 1.0.1
wheel 0.34.2
widgetsnbextension 3.5.1
wrapt 1.12.1
xlrd 1.2.0
zict 2.0.0
zipp 3.1.0