From 70c027e9efac8afaae5d82deaefcb07f811db0e6 Mon Sep 17 00:00:00 2001 From: Alex Waygood Date: Sun, 19 Feb 2023 15:22:51 +0100 Subject: [PATCH] Fix broken stubsabot logic in `find_first_release_with_py_typed` (#9760) --- scripts/stubsabot.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/scripts/stubsabot.py b/scripts/stubsabot.py index 8153fb11f..eb5621977 100644 --- a/scripts/stubsabot.py +++ b/scripts/stubsabot.py @@ -171,15 +171,15 @@ async def release_contains_py_typed(release_to_download: PypiReleaseDownload, *, async def find_first_release_with_py_typed(pypi_info: PypiInfo, *, session: aiohttp.ClientSession) -> PypiReleaseDownload | None: - release_iter = pypi_info.releases_in_descending_order() + release_iter = (release for release in pypi_info.releases_in_descending_order() if not release.version.is_prerelease) + latest_release = next(release_iter) # If the latest release is not py.typed, assume none are. - if not (await release_contains_py_typed(release := next(release_iter), session=session)): + if not (await release_contains_py_typed(latest_release, session=session)): return None - first_release_with_py_typed: PypiReleaseDownload | None = None + first_release_with_py_typed = latest_release while await release_contains_py_typed(release := next(release_iter), session=session): - if not release.version.is_prerelease: - first_release_with_py_typed = release + first_release_with_py_typed = release return first_release_with_py_typed