Tweak download script

This commit is contained in:
Matthew Honnibal 2016-10-19 00:52:16 +02:00
parent 5a5c7192a5
commit 622b0a9674
1 changed files with 7 additions and 4 deletions

View File

@ -4,11 +4,14 @@ from ..download import download
@plac.annotations(
force=("Force overwrite", "flag", "f", bool),
ignore_glove=("Don't install the GloVe vectors", "flag", "g", bool),
)
def main(data_size='all', force=False, ignore_glove=False):
download('en', force)
download('en_glove_cc_300_1m_vectors', force)
def main(data_size='all', force=False):
if data_size in ('all', 'parser'):
print("Downloading parsing model")
download('en', force)
if data_size in ('all', 'glove'):
print("Downloading GloVe vectors")
download('en_glove_cc_300_1m_vectors', force)
if __name__ == '__main__':