Stop dropping TypedDict annotations from local assignments.

Previously if we had
```
  f: SomeTypedDict = {...}
```
and f was not a module-level variable, we would drop the type (after checking
the dict's validity) and store it as a `Dict[str, ...]` instead.

Fixes #1185

PiperOrigin-RevId: 443447773

Author: martindemello

pytype/annotation_utils.py

@@ -10,7 +10,6 @@
 from pytype.abstract import abstract
 from pytype.abstract import abstract_utils
 from pytype.abstract import mixin
-from pytype.overlays import typed_dict
 from pytype.overlays import typing_overlay
 from pytype.pytd import pytd_utils
 
@@ -345,10 +344,6 @@ def apply_annotation(self, node, op, name, value):
       return AnnotatedValue(None, value, final=True)
     elif isinstance(typ, abstract.FinalAnnotation):
       return AnnotatedValue(typ.annotation, annot_val, final=True)
-    elif isinstance(typ, typed_dict.TypedDictClass):
-      # We do not want to instantiate a TypedDict annotation if we have a
-      # concrete value.
-      return AnnotatedValue(typ, value)
     elif typ.full_name == "typing.TypeAlias":
       # Validate that 'value' is a legal type alias and use it.
       annot = self.extract_annotation(node, value, name, stack)

pytype/tests/test_typed_dict.py

@@ -128,6 +128,40 @@ class A(TypedDict):
         "type errors", "{'x': ...}", "expected int", "got str"
     ]})
 
+  def test_annotated_global_var(self):
+    ty = self.Infer("""
+      from typing_extensions import TypedDict
+      class A(TypedDict):
+        x: int
+      a: A = {'x': 10}
+    """)
+    self.assertTypesMatchPytd(ty, """
+      from typing import TypedDict
+
+      class A(TypedDict):
+        x: int
+
+      a: A
+    """)
+
+  def test_annotated_local_var(self):
+    ty = self.Infer("""
+      from typing_extensions import TypedDict
+      class A(TypedDict):
+        x: int
+      def f():
+        a: A = {'x': 10}
+        return a
+    """)
+    self.assertTypesMatchPytd(ty, """
+      from typing import TypedDict
+
+      class A(TypedDict):
+        x: int
+
+      def f() -> A: ...
+    """)
+
   def test_return_type(self):
     err = self.CheckWithErrors("""
       from typing_extensions import TypedDict