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()