Colab notebook breaking after TF1 support removed

[cmgchess]

Description

The iPython notebook breaks as mentioned in the Issue title

# Error logs:
ValueError: Tensorflow 1 is unsupported in Colab.

Your notebook should be updated to use Tensorflow 2.
See the guide at https://www.tensorflow.org/guide/migrate#migrate-from-tensorflow-1x-to-tensorflow-2.

[Scherlac]

did you find some solution on this? is there a fork that solves this?