Fix unittest random failed of test_datasets (#30804)

* fix test_datasets unittest
5 years ago · 3a3ff75c52
parent 04532b8a83
commit 3a3ff75c52
1 changed files with 7 additions and 0 deletions
--- a/python/paddle/vision/datasets/flowers.py
+++ b/python/paddle/vision/datasets/flowers.py
@ -128,6 +128,13 @@ class Flowers(Dataset):

        scio = try_import('scipy.io')

+        # double check data download
+        self.label_file = _check_exists_and_download(self.label_file, LABEL_URL,
+                                                     LABEL_MD5, 'flowers', True)
+
+        self.setid_file = _check_exists_and_download(self.setid_file, SETID_URL,
+                                                     SETID_MD5, 'flowers', True)
+
        self.labels = scio.loadmat(self.label_file)['labels'][0]
        self.indexes = scio.loadmat(self.setid_file)[self.flag][0]