From 3375d48f8cdbee4d90bededb06d9aba0e6f6e987 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Wed, 29 Apr 2026 18:08:56 +0200 Subject: [PATCH] Improve typing_extensions usages --- jedi/inference/gradual/typeshed.py | 2 +- test/completion/pep0484_typing.py | 7 +++++++ 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/jedi/inference/gradual/typeshed.py b/jedi/inference/gradual/typeshed.py index 651f54e3..b981a6c0 100644 --- a/jedi/inference/gradual/typeshed.py +++ b/jedi/inference/gradual/typeshed.py @@ -282,7 +282,7 @@ def parse_stub_module(inference_state, file_io): def create_stub_module(inference_state, grammar, python_value_set, stub_module_node, file_io, import_names): - if import_names == ('typing',): + if import_names in [('typing',), ('typing_extensions',)]: module_cls = TypingModuleWrapper else: module_cls = StubModuleValue diff --git a/test/completion/pep0484_typing.py b/test/completion/pep0484_typing.py index d52da23a..e0004cd6 100644 --- a/test/completion/pep0484_typing.py +++ b/test/completion/pep0484_typing.py @@ -560,6 +560,8 @@ def typed_dict_test_foo(arg: Bar): # Self # ----------------- +import typing_extensions + # From #2023, #2068 class Builder: def __init__(self): @@ -577,6 +579,9 @@ class Builder: def add_not_implemented(self: Self, y: int) -> Self: raise NotImplementedError + def add_not_implemented_typing_extensions(self: Self, y: int) -> typing_extensions.Self: + raise NotImplementedError + b = Builder() #? Builder() b.add_x(2) @@ -584,3 +589,5 @@ b.add_x(2) b.add_x(2).add_y(5) #? Builder() b.add_x(2).add_not_implemented(5) +#? Builder() +b.add_x(2).add_not_implemented_typing_extensions(5)