DeepGraphLearning · wradstok · Jul 20, 2020 · Jul 20, 2020 · Jul 23, 2020
diff --git a/codes/run.py b/codes/run.py
@@ -123,7 +123,10 @@ def read_triple(file_path, entity2id, relation2id):
    triples = []
    with open(file_path) as fin:
        for line in fin:
-            h, r, t = line.strip().split('\t')
+            # The entity/relation dict have the element names at the end, when loading them
+            # the entire line is stripped, removing any trailing spaces. As such, we need
+            # to strip each element individually here as well.
+            h, r, t = map(str.strip, line.split('\t'))
            triples.append((entity2id[h], relation2id[r], entity2id[t]))
    return triples

@@ -160,12 +163,12 @@ def log_metrics(mode, step, metrics):

 def main(args):
    if (not args.do_train) and (not args.do_valid) and (not args.do_test):
-        raise ValueError('one of train/val/test mode must be choosed.')
+        raise ValueError('one of train/val/test mode must be chosen.')

    if args.init_checkpoint:
        override_config(args)
    elif args.data_path is None:
-        raise ValueError('one of init_checkpoint/data_path must be choosed.')
+        raise ValueError('one of init_checkpoint/data_path must be chosen.')

    if args.do_train and args.save_path is None:
        raise ValueError('Where do you want to save your trained model?')