Error on Federated_learning.ipynb

[Wapiti08]

Hi Author,

My python version is 3.7,
I was trying to run Federated Learning.ipynb on colab,
If I just downloaded the libraries as the following instructions,

pip install tensorflow_federated==0.2.0 tensorflow-datasets tensorflow==1.13.1

I got the problem like:

No module named 'tensorflow.compat.v2' in tensorflow_datasets

I also tried to run based on tf 2.x,
I also did not get a successful run. Of course, a lot of difference after I changed the tf version

Can you please specify the tensorflow_datasets version in this resp? Really appreciate that