Intellisense not working when using a specific import / return combination

Environment data

• VS Code version: 1.27.2
• Extension version (available under the Extensions sidebar): 2018.9.0
• OS and version: Ubuntu 18.04.1 LTS
• Python version (& distribution if applicable, e.g. Anaconda): Python 3.6.6
• Type of virtual environment used (N/A | venv | virtualenv | conda | …): virtualenv
• Relevant/affected Python packages and their versions: /

Actual behavior

Inellisense not working when using a specific import / return combination.

Expected behavior

Intellisense to work.

Steps to reproduce:

File policy/ab.py:

class BaseAB:
    def smthng(self, i):
        pass

class A(BaseAB):
    def smthng(self, i):
        print("A smthng " + str(i))

class B(BaseAB):
    def smthng(self, i):
        print("B smthng " + str(i))

File policy/__init__.py:

from policy.ab import A, B

def build_test(input):
    return {
        'A': A,
        'B': B
    }[input]()

Intellisense does not know what build_test returns:

Interestingly, it works in other combinations, like the same input but when return is with ifs: Or the same return but when import is a bit different, when not importing to the class level:

[brettcannon]

[ @bjg2 I edited your ✉️ ⬆️ to come off as more 😃 and less 😠 ]

We are just swamped with things, but we will triage this and figure out who to reach out to in order to get this potentially fixed.

[kimadeline]

Closing, language server & jedi have the same result for all combinations, and go to definition works.