Make sure that typeddict py__getitem__ works

2020-02-07 16:45:03 +01:00
parent 7ef07b576f
commit 87161df2f0
4 changed files with 17 additions and 10 deletions
--- a/jedi/inference/gradual/typing.py
+++ b/jedi/inference/gradual/typing.py
@@ -389,7 +389,13 @@ class TypedDict(LazyValueWrapper):
        return ValueName(self, self.tree_node.name)

    def py__simple_getitem__(self, index):
-        return ValueSet({self.inference_state.builtins_module})
+        if isinstance(index, str):
+            return ValueSet.from_sets(
+                name.infer()
+                for filter in self._definition_class.get_filters(is_instance=True)
+                for name in filter.get(index)
+            )
+        return NO_VALUES

    def get_key_values(self):
        from jedi.inference.compiled import create_simple_object