From 03badaff57da23a80d739459cc159adf94cbd2ef Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 16 Apr 2024 16:19:31 -0700 Subject: [PATCH] Improve PNNX assets check robustness (#10089) --- ultralytics/engine/exporter.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index d5addaaf..d3873843 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -530,8 +530,9 @@ class Exporter: ) system = "macos" if MACOS else "windows" if WINDOWS else "linux-aarch64" if ARM64 else "linux" try: - _, assets = get_github_assets(repo="pnnx/pnnx", retry=True) + _, assets = get_github_assets(repo="pnnx/pnnx") url = [x for x in assets if f"{system}.zip" in x][0] + assert url, "Unable to retrieve PNNX repo assets" except Exception as e: url = f"https://github.com/pnnx/pnnx/releases/download/20240410/pnnx-20240410-{system}.zip" LOGGER.warning(f"{prefix} WARNING ⚠️ PNNX GitHub assets not found: {e}, using default {url}")