legacy notebooks broken since tensorflow is officially deprecated in colab, just fwiw