From 971d4ba141459f78d10d5770b9459d1ead7d49a0 Mon Sep 17 00:00:00 2001
From: Scott <scott@wannabe.house>
Date: Wed, 2 Feb 2022 08:31:42 -0800
Subject: More mypy cleanup.

---
 ansi.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'ansi.py')

diff --git a/ansi.py b/ansi.py
index 9e31b81..6897291 100755
--- a/ansi.py
+++ b/ansi.py
@@ -1890,19 +1890,19 @@ def bg(
 
 class StdoutInterceptor(io.TextIOBase):
     def __init__(self):
-        self.saved_stdout: Optional[io.TextIOBase] = None
+        self.saved_stdout: io.TextIO = None
         self.buf = ''
 
     @abstractmethod
     def write(self, s: str):
         pass
 
-    def __enter__(self) -> None:
+    def __enter__(self):
         self.saved_stdout = sys.stdout
         sys.stdout = self
-        return None
+        return self
 
-    def __exit__(self, *args) -> bool:
+    def __exit__(self, *args) -> Optional[bool]:
         sys.stdout = self.saved_stdout
         print(self.buf)
         return None
-- 
cgit v1.3