From 504949c6b97662a9deff676452cc15d146b35c7c Mon Sep 17 00:00:00 2001 From: Prabhat Roy Date: Thu, 20 May 2021 02:51:04 -0700 Subject: [PATCH] Added url to class mapping logic to manifold download Reviewed By: fmassa Differential Revision: D28537606 fbshipit-source-id: beaf15f4e38a7d986ef26587383aa2aee3e48952 --- torchvision/datasets/_utils.py | 2 +- torchvision/datasets/utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvision/datasets/_utils.py b/torchvision/datasets/_utils.py index d0664d50d54..5661b31a4b9 100644 --- a/torchvision/datasets/_utils.py +++ b/torchvision/datasets/_utils.py @@ -1,4 +1,4 @@ -def _download_file_from_remote_location(fpath: str) -> None: +def _download_file_from_remote_location(fpath: str, url: str) -> None: pass diff --git a/torchvision/datasets/utils.py b/torchvision/datasets/utils.py index 36d6f8e657e..e6798bd4500 100644 --- a/torchvision/datasets/utils.py +++ b/torchvision/datasets/utils.py @@ -123,7 +123,7 @@ def download_url( return if _is_remote_location_available(): - _download_file_from_remote_location(fpath) + _download_file_from_remote_location(fpath, url) else: # expand redirect chain if needed url = _get_redirect_url(url, max_hops=max_redirect_hops)