From 21fa00654a582ad2a659ee2a35fc5046275c9cff Mon Sep 17 00:00:00 2001 From: F-G Fernandez Date: Fri, 18 Oct 2019 12:04:13 +0200 Subject: [PATCH] test: Updated assert in test_hub Updated all raw asserts to corresponding unittest.TestCase.assert. See #1483 --- test/test_hub.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_hub.py b/test/test_hub.py index 2f0ddfb2537..4ae9e51021b 100644 --- a/test/test_hub.py +++ b/test/test_hub.py @@ -44,7 +44,7 @@ def test_set_dir(self): progress=False) self.assertEqual(sum_of_model_parameters(hub_model).item(), SUM_OF_PRETRAINED_RESNET18_PARAMS) - assert os.path.exists(temp_dir + '/pytorch_vision_master') + self.assertTrue(os.path.exists(temp_dir + '/pytorch_vision_master')) shutil.rmtree(temp_dir + '/pytorch_vision_master') def test_list_entrypoints(self):