Fixed URLContext Leak (#1429)

488f36b4 · Mustafa Bal · GitHub · 6d5e0ba6 · 488f36b4
Unverified Commit 488f36b4 authored Apr 06, 2021 by Mustafa Bal Committed by GitHub Apr 06, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

torchaudio/datasets/utils.py torchaudio/datasets/utils.py +2 -1

No files found.
--- a/torchaudio/datasets/utils.py
+++ b/torchaudio/datasets/utils.py
@@ -29,7 +29,8 @@ def stream_url(url: str,

    # If we already have the whole file, there is no need to download it again
    req = urllib.request.Request(url, method="HEAD")
-    url_size = int(urllib.request.urlopen(req).info().get("Content-Length", -1))
+    with urllib.request.urlopen(req) as response:
+        url_size = int(response.info().get("Content-Length", -1))
    if url_size == start_byte:
        return