From a36abd4297a878d5ba7fc247daede03d7495c2d4 Mon Sep 17 00:00:00 2001 From: SkalskiP Date: Tue, 24 Oct 2023 14:21:58 +0200 Subject: [PATCH] "Fix reference bug in video asset downloader Updated 'asset_name' to 'filename' in downloader.py. The downloader was using the incorrect variable 'asset_name', causing failed video asset downloads. Correcting to 'filename' allows for accurate md5 check and download." --- supervision/assets/downloader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/supervision/assets/downloader.py b/supervision/assets/downloader.py index 6ef53c4a..6ce25b00 100644 --- a/supervision/assets/downloader.py +++ b/supervision/assets/downloader.py @@ -79,10 +79,10 @@ def download_assets(asset_name: Union[VideoAssets, str]) -> str: copyfileobj(r_raw, f) elif path_exists(filename): - if not is_md5_hash_matching(filename, VIDEO_ASSETS[asset_name][2]): + if not is_md5_hash_matching(filename, VIDEO_ASSETS[filename][2]): print("File corrupted. Re-downloading... \n") os_remove(filename) - download_assets(asset_name) + download_assets(filename) print(f"{filename} asset download complete. \n") else: