"src/vscode:/vscode.git/clone" did not exist on "5af9aac00a02daa1f2c351bcdc0063e00f30d89a"
Unverified Commit f5f60934 authored by Leo Gao's avatar Leo Gao Committed by GitHub
Browse files

Merge pull request #129 from jon-tow/write-out-fix

Fix `write_out.py` training set bug
parents c0fbf9e8 990d5e31
...@@ -37,8 +37,8 @@ def main(): ...@@ -37,8 +37,8 @@ def main():
iters = [] iters = []
for set in args.sets.split(","): for set in args.sets.split(","):
if set == 'train' and task.has_train_docs(): if set == 'train' and task.has_training_docs():
docs = task.train_docs() docs = task.training_docs()
if set == 'val' and task.has_validation_docs(): if set == 'val' and task.has_validation_docs():
docs = task.validation_docs() docs = task.validation_docs()
if set == 'test' and task.has_test_docs(): if set == 'test' and task.has_test_docs():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment