Jupyter NotebookでのKerasモデルの表示にAssertionErrorが出る

Question

東大松尾研から一般公開されているdl4usのlesson1_sec1においてKerasでグラフを表示しようとするとAssertionErrorが起き表示できません。

https://weblab.t.u-tokyo.ac.jp/dl4us/

発生している問題・エラーメッセージ

AssertionError                            Traceback (most recent call last)
<ipython-input-11-6687659cb501> in <module>()
      2 from tensorflow.python.keras.utils.vis_utils import model_to_dot
      3 
----> 4 SVG(model_to_dot(model, show_shapes=True).create(prog='dot', format='svg'))

/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/utils/vis_utils.py in model_to_dot(model, show_shapes, show_dtype, show_layer_names, rankdir, expand_nested, dpi, subgraph)
    236           inbound_layer_id = str(id(inbound_layer))
    237           if not expand_nested:
--> 238             assert dot.get_node(inbound_layer_id)
    239             assert dot.get_node(layer_id)
    240             add_edge(dot, inbound_layer_id, layer_id)

AssertionError: 

該当のソースコード

from IPython.display import SVG
from tensorflow.python.keras.utils.vis_utils import model_to_dot

SVG(model_to_dot(model, show_shapes=True).create(prog='dot', format='svg'))

試したこと

pydotplut、graphvizは導入しました。
graphvizについてはPATHが通っていることも確認済みです。
google colab、ローカルでのjupyter notebookどちらについても同じエラーが出ます。

どなたかご教示くださると嬉しいです。

Answer 1

lesson1_sec1_exercise.ipynb
をGoogle Colabにアップロードして、先頭に

!pip install tensorflow==2.4.3
!pip install keras==2.4.3

を追加して実行したら、エラー出ませんでした