Decorator seems not to be applied to a function under some circumstances (inferred return type, recursive reference, loop, generator)

[Azureblade3808]

Describe the bug

I am not able to describe it clearly. See sample code below.

Code or Screenshots

from __future__ import annotations

from contextlib import contextmanager

@contextmanager
# The following function is having an omitted return type.
# If the return type is declared explicitly (even as `-> Any`), this issue goes away.
def f():
    yield

    # Must be inside a loop. `for` or `while` doesn't matter.
    for _ in ():
        # The following `f()` is regarded as if it's having its original signature and 
        # returns a `Generator` instead of a context manager, so the `with` clause gets
        # reported.
        with f():
            return

VS Code extension or command-line

Pyright 1.1.389.

[erictraut]

This is addressed in pyright 1.1.390.