diff options
| author | John Bauer <horatio@gmail.com> | 2022-09-10 01:12:04 +0300 |
|---|---|---|
| committer | John Bauer <horatio@gmail.com> | 2022-09-10 01:12:04 +0300 |
| commit | ec3e731a35be41886c24bf508727ab494588ebb7 (patch) | |
| tree | 1db9140bfa2cc13dcad889542dfc540b2e51a0e6 | |
| parent | 1548568455a13cb2c37ad1cf053225ee068f63b0 (diff) | |
Technically it works the old way, but the filenames look silly
| -rw-r--r-- | stanza/tests/constituency/test_trainer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/stanza/tests/constituency/test_trainer.py b/stanza/tests/constituency/test_trainer.py index ad94ed22..1e05f249 100644 --- a/stanza/tests/constituency/test_trainer.py +++ b/stanza/tests/constituency/test_trainer.py @@ -163,7 +163,7 @@ class TestTrainer: save_name = os.path.join(args['save_dir'], args['save_name']) latest_name = os.path.join(args['save_dir'], 'latest.pt') - each_name = os.path.join(args['save_dir'], 'each_%2d.pt') + each_name = os.path.join(args['save_dir'], 'each_%02d.pt') assert not os.path.exists(save_name) retag_pipeline = Pipeline(lang="en", processors="tokenize, pos", tokenize_pretokenized=True) tr = trainer.train(args, save_name, None, latest_name, each_name, retag_pipeline) @@ -212,7 +212,7 @@ class TestTrainer: if use_lattn: args += ['--lattn_d_proj', '16'] args = self.run_train_test(wordvec_pretrain_file, tmpdirname, num_epochs=8, extra_args=args) - each_name = os.path.join(args['save_dir'], 'each_%2d.pt') + each_name = os.path.join(args['save_dir'], 'each_%02d.pt') word_input_sizes = defaultdict(list) for i in range(1, 9): |