From 440cb9d2bca367472d56a3b0fe4e649cc9c19ab4 Mon Sep 17 00:00:00 2001
From: Peter Rowlands <peter@pmrowla.com>
Date: Fri, 17 Mar 2023 14:13:34 +0900
Subject: [PATCH] ui: flush stdout/stderr before colorama re-init

---
 dvc/ui/__init__.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/dvc/ui/__init__.py b/dvc/ui/__init__.py
index 0fc60bceb..add18faf6 100644
--- a/dvc/ui/__init__.py
+++ b/dvc/ui/__init__.py
@@ -31,10 +31,16 @@ if TYPE_CHECKING:
 
 @contextmanager
 def disable_colorama():
+    import sys
+
     colorama.deinit()
     try:
         yield
     finally:
+        if sys.stdout:
+            sys.stdout.flush()
+        if sys.stderr:
+            sys.stderr.flush()
         colorama.reinit()