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,10 +4,13 @@ from ..download import download
@plac.annotations( @plac.annotations(
force=("Force overwrite", "flag", "f", bool), 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): def main(data_size='all', force=False):
if data_size in ('all', 'parser'):
print("Downloading parsing model")
download('en', force) download('en', force)
if data_size in ('all', 'glove'):
print("Downloading GloVe vectors")
download('en_glove_cc_300_1m_vectors', force) download('en_glove_cc_300_1m_vectors', force)