Added _compat typing.

pallets · Nov 9, 2024 · 63db9af · 63db9af
1 parent 8d3eabd
commit 63db9af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/click/_compat.py b/src/click/_compat.py
@@ -538,14 +538,14 @@ def auto_wrap_for_ansi(stream: t.TextIO, color: bool | None = None) -> t.TextIO:
         rv = t.cast(t.TextIO, ansi_wrapper.stream)
         _write = rv.write
 
-        def _safe_write(s):
+        def _safe_write(s: str) -> int:
             try:
                 return _write(s)
             except BaseException:
                 ansi_wrapper.reset_all()
                 raise
 
-        rv.write = _safe_write
+        rv.write = _safe_write  # type: ignore[method-assign]
 
         try:
             _ansi_stream_wrappers[stream] = rv