Fix TODOs in binder.py

python · Mar 28, 2017 · 0f06e06 · 0f06e06
1 parent 8e385f0
commit 0f06e06
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/mypy/binder.py b/mypy/binder.py
@@ -110,8 +110,7 @@ def _get(self, key: Key, index: int=-1) -> Type:
         return None
 
     def put(self, expr: Expression, typ: Type) -> None:
-        # TODO: replace with isinstance(expr, BindableTypes)
-        if not isinstance(expr, (IndexExpr, MemberExpr, NameExpr)):
+        if not isinstance(expr, BindableTypes):
             return
         if not expr.literal:
             return
@@ -203,8 +202,7 @@ def assign_type(self, expr: Expression,
                     type: Type,
                     declared_type: Type,
                     restrict_any: bool = False) -> None:
-        # TODO: replace with isinstance(expr, BindableTypes)
-        if not isinstance(expr, (IndexExpr, MemberExpr, NameExpr)):
+        if not isinstance(expr, BindableTypes):
             return None
         if not expr.literal:
             return