Colab example doesn't run successfuly

[agemagician]

Hello,

I am trying to run the jupyter notebook example "introduction.ipynb" on "codelabs" folder using Google Colab. However, it gives me the following error:

2021-05-12 03:12:03.347414: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcudart.so.10.1
WARNING:absl:Lingvo does not support eager execution yet. Please disable eager execution with tf.compat.v1.disable_eager_execution() or proceed at your own risk.
Traceback (most recent call last):
  File "/usr/lib/python3.7/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/usr/lib/python3.7/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/usr/local/lib/python3.7/dist-packages/lingvo/trainer.py", line 39, in <module>
    from lingvo import executor
  File "/usr/local/lib/python3.7/dist-packages/lingvo/executor.py", line 21, in <module>
    from lingvo.core import base_model
  File "/usr/local/lib/python3.7/dist-packages/lingvo/core/base_model.py", line 35, in <module>
    from model_pruning.python import pruning
  File "/usr/local/lib/python3.7/dist-packages/model_pruning/__init__.py", line 37, in <module>
    from model_pruning.python.pruning_interface import apply_matrix_compression
  File "/usr/local/lib/python3.7/dist-packages/model_pruning/python/pruning_interface.py", line 44, in <module>
    from graph_compression.compression_lib import compression_wrapper_py2 as compression_wrapper
ModuleNotFoundError: No module named 'graph_compression'

When I try to run the "Model Training" block.

Any ideas how could we fix this problem ?