Wow I copied the example from the collab to my local python REPL and it prints the same tensorflow version but still fails with the error from the original post. I cannot reproduce this error on collab