Skip to content

Commit

Permalink
fix(cli): avoid false positive cli upgrade suggestions (#12497)
Browse files Browse the repository at this point in the history
  • Loading branch information
hsheth2 authored Jan 31, 2025
1 parent ec0eb06 commit 317b740
Showing 1 changed file with 13 additions and 5 deletions.
18 changes: 13 additions & 5 deletions metadata-ingestion/src/datahub/upgrade/upgrade.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,11 +55,19 @@ async def get_client_version_stats():
async with session.get(pypi_url) as resp:
response_json = await resp.json()
try:
releases = response_json.get("releases", [])
sorted_releases = sorted(releases.keys(), key=lambda x: Version(x))
latest_cli_release_string = [
x for x in sorted_releases if "rc" not in x
][-1]
releases = response_json.get("releases", {})
filtered_releases = {
version: release_files
for version, release_files in releases.items()
if not all(
release_file.get("yanked") for release_file in release_files
)
and "rc" not in version
}
sorted_releases = sorted(
filtered_releases.keys(), key=lambda x: Version(x)
)
latest_cli_release_string = sorted_releases[-1]
latest_cli_release = Version(latest_cli_release_string)
current_version_info = releases.get(current_version_string)
current_version_date = None
Expand Down

0 comments on commit 317b740

Please sign in to comment.