diff --git a/stanza/tests/constituency/test_trainer.py b/stanza/tests/constituency/test_trainer.py index fe660a89d..0b4cecae0 100644 --- a/stanza/tests/constituency/test_trainer.py +++ b/stanza/tests/constituency/test_trainer.py @@ -212,7 +212,7 @@ def run_train_test(self, wordvec_pretrain_file, tmpdirname, num_epochs=5, extra_ each_name = args['save_each_name'] if not exists_ok: assert not os.path.exists(args['save_name']) - retag_pipeline = Pipeline(lang="en", processors="tokenize, pos", tokenize_pretokenized=True, dir=TEST_MODELS_DIR, foundation_cache=foundation_cache) + retag_pipeline = Pipeline(lang="en", processors="tokenize, pos", tokenize_pretokenized=True, dir=TEST_MODELS_DIR, foundation_cache=foundation_cache, download_method=None) trained_model = parser_training.train(args, None, [retag_pipeline]) # check that hooks are in the model if expected for p in trained_model.model.parameters():