Skip to content

Commit

Permalink
don't redownload the resources for every test
Browse files Browse the repository at this point in the history
  • Loading branch information
AngledLuffa committed Feb 23, 2025
1 parent 04f03d9 commit fd0f94c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion stanza/tests/constituency/test_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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():
Expand Down

0 comments on commit fd0f94c

Please sign in to comment.