Skip to content

Commit b5c4641

Browse files
authored
Add error handling for staleness report generation (#648)
1 parent 0a453c5 commit b5c4641

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/package_report.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,12 +87,19 @@ def _generate_staleness_report_per_image(
8787
if version_in_sagemaker_distribution == package_versions_in_upstream[package]
8888
else "${\color{red}" + package + "}$"
8989
)
90+
91+
# Get download count with error handling
92+
try:
93+
download_count = conda_download_stats[package]
94+
except (KeyError, TypeError):
95+
download_count = 0
96+
9097
staleness_report_rows.append(
9198
{
9299
"package": package_string,
93100
"version_in_sagemaker_distribution": version_in_sagemaker_distribution,
94101
"latest_relavant_version": package_versions_in_upstream[package],
95-
"downloads": conda_download_stats[package],
102+
"downloads": download_count,
96103
}
97104
)
98105

0 commit comments

Comments
 (0)