Always check for state

python · Jan 12, 2019 · ae08bd3 · ae08bd3
1 parent 860de83
commit ae08bd3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mypy/stubutil.py b/mypy/stubutil.py
@@ -162,7 +162,8 @@ def infer_sig_from_docstring(docstr: str, name: str) -> Optional[List[TypedFunct
                 # reset state, function name not followed by '('
                 state.pop()
 
-            elif token.type == tokenize.OP and token.string in ('[', '(', '{'):
+            elif token.type == tokenize.OP and token.string in ('[', '(', '{') and \
+                    state[-1] != State.INIT:
                 accumulator += token.string
                 state.append(State.OPEN_BRACKET)
 
@@ -206,7 +207,7 @@ def infer_sig_from_docstring(docstr: str, name: str) -> Optional[List[TypedFunct
                 arg_default = None
                 accumulator = ""
 
-            elif token.type == tokenize.OP and token.string == '->':
+            elif token.type == tokenize.OP and token.string == '->' and state[-1] == State.INIT:
                 accumulator = ""
                 state.append(State.RETURN_VALUE)