Fix for dataloader (dmlc#1353)

skorani · Mar 12, 2020 · 87842db · 87842db
1 parent 3938b39
commit 87842db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/apps/kg/dataloader/KGDataset.py b/apps/kg/dataloader/KGDataset.py
@@ -272,7 +272,7 @@ def read_relation(self, relation_path):
             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 = []