Parameter suggestion uses only the last overload for overloaded functions

[adrien-lsh]

Hi there,

I recently started to type hint the functions of my project. And some of them return certain types depending of the value of a parameter.
When overloading these functions to have accurate types, I found out that my lsp only suggests one of the two possibilities.

Pylance.issue.mp4

Python 3.12.2
Neovim v0.10.4
Pyright 1.1.393

[rchiodo]

Seems like a bug to me.

Here's the code typed out:

@overload
def foo(return_type: Literal['dict']) -> dict: ...  
@overload
def foo(return_type: Literal['list']) -> list: ...

def foo(return_type: Literal['dict', 'list']) -> dict | list:
    if return_type == 'dict':
        return {}
    return []

foo('')

[adrien-lsh]

Thank you !