Tweak download script

This commit is contained in:
Matthew Honnibal 2016-10-19 00:52:16 +02:00
parent 5a5c7192a5
commit 622b0a9674

View File

@ -4,11 +4,14 @@ 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):
download('en', force) if data_size in ('all', 'parser'):
download('en_glove_cc_300_1m_vectors', force) 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__': if __name__ == '__main__':