PT022 doesnt correct type annotations

[RonnyPfannschmidt]

given

import pytest
from typing import Generator


@pytest.fixture()
def testme() -> Generator[int, None, None]:
    a = 1
    yield a

after running

ruff --isolated --select PT ruff_check.py --fix

im left with

import pytest
from typing import Generator


@pytest.fixture()
def testme() -> Generator[int, None, None]:
    a = 1
    return a

instead of

import pytest
from typing import Generator


@pytest.fixture()
def testme() -> int:
    a = 1
    return a

[charliermarsh]

Thanks for filing. Just confirming, the issue is that the type annotation should be changed to reflect that the method is no longer a generator, right?

[RonnyPfannschmidt]

exactly