message: Forces interactive python printing in TF to flush at the end of a print. This will hopefully fix buffering issues in some colab environments.
diff: @@ -3147,6 +3147,10 @@ void PrintToPythonStdout(const char* msg) {
  }
  PySys_WriteStdout("\n");
  + // Force flushing to make sure print newlines aren't interleaved in
  + // some colab environments
  + PyRun_SimpleString("import sys; sys.stdout.flush()");
  +
  PyGILState_Release(py_threadstate);
  }
  }
  