fix caching ground truth

Thinklab-SJTU · Jun 20, 2024 · 2dc7d29 · 2dc7d29
1 parent 802abd2
commit 2dc7d29
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pygmtools/benchmark.py b/pygmtools/benchmark.py
@@ -187,8 +187,7 @@ def get_data(self, ids, test=False, shuffle=True):
             for pair in id_combination:
                 id_pair = (ids[pair[0]], ids[pair[1]])
                 gt_path = os.path.join(self.gt_cache_path, str(id_pair) + '.npy')
-                if not os.path.exists(gt_path):
-                    np.save(gt_path, perm_mat_dict[pair])
+                np.save(gt_path, perm_mat_dict[pair])
 
         if not test:
             return data_list, perm_mat_dict, ids