Fix return value check for subprocess.run (ultralytics#10972)

Subprocess.run does not return an integer. Regressed in ultralytics#10944
sanweiw · Feb 13, 2023 · 1a2eb53 · 1a2eb53
1 parent fa4bdbe
commit 1a2eb53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/general.py b/utils/general.py
@@ -631,8 +631,8 @@ def download_one(url, dir):
             for i in range(retry + 1):
                 if curl:
                     s = 'sS' if threads > 1 else ''  # silent
-                    r = subprocess.run(f'curl -# -{s}L "{url}" -o "{f}" --retry 9 -C -'.split())
-                    success = r == 0
+                    proc = subprocess.run(f'curl -# -{s}L "{url}" -o "{f}" --retry 9 -C -'.split())
+                    success = proc.returncode == 0
                 else:
                     torch.hub.download_url_to_file(url, f, progress=threads == 1)  # torch download
                     success = f.is_file()