stop-iteration-return false positive when using itertools.cycle

[richardfearn]

Bug description

Running pylint on this code gives a false positive:

from itertools import cycle


def test():
    vals = cycle('ABCD')
    while True:
        yield next(vals)

This is similar to #2158 which was fixed by adding a list of known infinite iterators.

Configuration

No response

Command used

pylint test.py

Pylint output

************* Module test
test.py:7:14: R1708: Do not raise StopIteration in generator, use return statement instead (stop-iteration-return)

Expected behavior

No warning, because cycle repeats indefinitely.

Pylint version

pylint 2.12.2
astroid 2.9.3
Python 3.10.7 (main, Nov  2 2022, 18:49:29) [GCC 12.2.0]

OS / Environment

Ubuntu 22.10

Additional dependencies

No response

[richardfearn]

Thanks for the quick fix!