diff --git a/test/test_datasets_utils.py b/test/test_datasets_utils.py index 1940c9e8493..43fdbe8239b 100644 --- a/test/test_datasets_utils.py +++ b/test/test_datasets_utils.py @@ -7,6 +7,7 @@ import tarfile import gzip import warnings +from torch._six import PY2 from torch._utils_internal import get_file_path_2 from common_utils import get_tmp_dir @@ -40,6 +41,7 @@ def test_check_integrity(self): self.assertTrue(utils.check_integrity(existing_fpath)) self.assertFalse(utils.check_integrity(nonexisting_fpath)) + @unittest.skipIf(PY2, "https://github.com/pytorch/vision/issues/1268") def test_download_url(self): with get_tmp_dir() as temp_dir: url = "http://github.com/pytorch/vision/archive/master.zip" @@ -51,6 +53,7 @@ def test_download_url(self): warnings.warn(msg, RuntimeWarning) raise unittest.SkipTest(msg) + @unittest.skipIf(PY2, "https://github.com/pytorch/vision/issues/1268") def test_download_url_retry_http(self): with get_tmp_dir() as temp_dir: url = "https://github.com/pytorch/vision/archive/master.zip"