temp test fixes

python · Aug 2, 2023 · 3073175 · 3073175
1 parent 859f44a
commit 3073175
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test-data/unit/stubgen.test b/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