docs: def disable_stdout_buffering():
      """This turns off stdout buffering so that outputs are immediately
      materialized and log messages show up before the program exits"""
      stdout_orig = sys.stdout
      sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 0)
      # NOTE(brandyn): This removes the original stdout
      return stdout_orig