Tensorflow Federated Tutorial In Google Colab Giving Errors In The Initialization Code Snippet