Fix for dataloader (#1353)

87842db4 · xiang song(charlie.song) · GitHub · 3938b399 · 87842db4
Unverified Commit 87842db4 authored Mar 12, 2020 by xiang song(charlie.song) Committed by GitHub Mar 12, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

apps/kg/dataloader/KGDataset.py apps/kg/dataloader/KGDataset.py +1 -1

No files found.
--- a/apps/kg/dataloader/KGDataset.py
+++ b/apps/kg/dataloader/KGDataset.py
@@ -272,7 +272,7 @@ class KGDatasetFreebase(KGDataset):
            n_relations = int(f_rel.readline()[:-1])
        return None, n_relations

-    def read_triple(self, path, mode, skip_first_line=False):
+    def read_triple(self, path, mode, skip_first_line=False, format=None):
        heads = []
        tails = []
        rels = []