Fix ground truth check bug 0329

Manu87DS · Mar 29, 2023 · cb50be8 · cb50be8
1 parent c9ba4e5
commit cb50be8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/uie_dataset.py b/src/uie_dataset.py
@@ -315,6 +315,7 @@ def load_RE_dataset(self, dataset_path, labels_path, dataset_name, sampling_stra
                     ground_truth_pairs.append([relation['head']['name'], 'NA', relation['tail']['name']])
                     continue
                 relation_pair = [relation['head']['name'], relation['type'], relation['tail']['name']]
+                ground_truth_pairs.append(relation_pair)
                 relation_pairs.append(relation_pair)
 
             if len(relation_pairs) > 0:
@@ -325,6 +326,7 @@ def load_RE_dataset(self, dataset_path, labels_path, dataset_name, sampling_stra
             if len(ground_truth_pairs) > 0:
                 ground_truth = ", ".join(["({}, {}, {})".format(h, r, t) for (h, r, t) in ground_truth_pairs])
             else:
+                logger.error("******Error item: {}******".format(instance))
                 raise Exception('Dataset Error:{}, No ground truth!'.format(dataset_name))
 
             example["Instance"] = {