temp test fixes

Author: chadrik

test-data/unit/stubgen.test

@@ -3155,7 +3155,8 @@ def f(*args: Union[int, Tuple[int, int]]) -> int:
 
 
 [out]
-from typing import Tuple, overload
+from typing import Tuple
+from typing_extensions import overload
 
 class A:
     @overload
@@ -3514,7 +3515,7 @@ from typing import TypedDict
 X = TypedDict('X', a=int, b=str)
 Y = TypedDict('X', a=int, b=str, total=False)
 [out]
-from typing import TypedDict
+from typing_extensions import TypedDict
 
 class X(TypedDict):
     a: int