Skip to content

Commit f35f379

Browse files
authored
Merge pull request #203 from Mayo42/master
Fixed missing custom_dataset_path parameter
2 parents 6b081a1 + 2667ed8 commit f35f379

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/pykg2vec_train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
def main():
1212
args = KGEArgParser().get_args(sys.argv[1:])
1313

14-
knowledge_graph = KnowledgeGraph(dataset=args.dataset_name)
14+
knowledge_graph = KnowledgeGraph(dataset=args.dataset_name, custom_dataset_path=args.dataset_path)
1515
knowledge_graph.prepare_data()
1616

1717
config_def, model_def = Importer().import_model_config(args.model_name.lower())

0 commit comments

Comments
 (0)