graph_nets/demos/shortest_path.ipynb TensorFlow version 1 is not supported in Google Colab anymore, so the code is not running.