Failure with Tensorflow >= 2.3

[kelsey-sorrels]

Calling plot_model(...) using tensorflow 2.3 or greater results in the error message

  File "/[redacted]/lib/python3.6/site-packages/plot_model/plot_model.py", line 91, in model_to_dot
    from tensorflow.python.keras.engine import network
ImportError: cannot import name 'network'

The network module is no longer present, see https://github.com/tensorflow/tensorflow/tree/v2.3.1/tensorflow/python/keras/engine

[kelsey-sorrels]

Got this working by replacing network and network.Network with functional and functional.Functional. Breaking change appears to be tensorflow/tensorflow@bb15c97#diff-7c9eaab01e9c5c241737b86bcdcca7ba6fcac13759cca29c39b1b6231ddc66cd

[CometManAtGitHub]

Hi.
Thanks.
This works with additional change for Tensorflow-2.5.0 on Windows.
layers = model.layers #_layers

in file:
plot_model.py line 127