ModuleNotFoundError: No module named 'tensorflow.compat.v2'

Description

I ran the following code in a Google Colab and get the error:

ModuleNotFoundError: No module named 'tensorflow.compat.v2'

!pip install -q tensorflow==1.13.1
!pip install -q tensorflow_probability==0.6.0
!pip install -q tensor2tensor==1.13.1
!pip install -q gym[atari]
!python -m tensor2tensor.rl.trainer_model_based \
  --loop_hparams_set=rlmb_base_stochastic_discrete \
  --loop_hparams=game=pong,epochs=1,ppo_epochs_num=0 \
  --output_dir=~/t2t_train/mb_sd_pong_random

$ pip freeze | grep tensor

mesh-tensorflow==0.1.16
tensor2tensor==1.13.1
tensorboard==1.13.1
tensorboard-plugin-wit==1.6.0.post3
tensorboardcolab==0.0.22
tensorflow==1.13.1
tensorflow-addons==0.8.3
tensorflow-datasets==2.1.0
tensorflow-estimator==1.13.0
tensorflow-gcs-config==2.2.0
tensorflow-hub==0.8.0
tensorflow-metadata==0.22.2
tensorflow-privacy==0.2.2
tensorflow-probability==0.6.0

$ python -V

Python 3.6.9