Skip to content

GitLab

Commit 990d5e31 authored by Jonathan Tow's avatar Jonathan Tow
Browse files

Fix `write_out.py` training set bug

parent c0fbf9e8
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
scripts/write_out.py
View file @ 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