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():
iters = []
for set in args.sets.split(","):
if set == 'train' and task.has_train_docs():
docs = task.train_docs()
if set == 'train' and task.has_training_docs():
docs = task.training_docs()
if set == 'val' and task.has_validation_docs():
docs = task.validation_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