Add type annotations

sebastinas · sebastinas · commit de90c373930f · 2021-05-16T22:14:05.000+02:00
diff --git a/bpython/config.py b/bpython/config.py
@@ -37,7 +37,7 @@
 
 
 class UnknownColorCode(Exception):
-    def __init__(self, key, color):
+    def __init__(self, key: str, color: str) -> None:
         self.key = key
         self.color = color
 
diff --git a/bpython/inspection.py b/bpython/inspection.py
@@ -29,6 +29,7 @@
 
 from pygments.token import Token
 from pygments.lexers import Python3Lexer
+from typing import Any
 from types import MemberDescriptorType
 
 from .lazyre import LazyReCompile
@@ -53,7 +54,7 @@ class AttrCleaner:
     """A context manager that tries to make an object not exhibit side-effects
     on attribute lookup."""
 
-    def __init__(self, obj):
+    def __init__(self, obj: Any) -> None:
         self.obj = obj
 
     def __enter__(self):
