fixed bug where reference transcriptions weren't written.

persephone-tools · Nov 13, 2019 · da67e98 · da67e98
1 parent 0cf1d4d
commit da67e98
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/changelog.md b/changelog.md
@@ -9,6 +9,7 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 ## [0.4.1] - 2019-11-13
 
 - Bugfix for an assertion that checked for equality of floating points.
+- Bugfix where reference transcription wasn't written.
 
 ## [0.4.0] - 2019-08-31
 

diff --git a/persephone/model.py b/persephone/model.py
@@ -421,7 +421,7 @@ def train(self, *, early_stopping_steps: int = 10, min_epochs: int = 30,
                               "w", encoding=ENCODING) as hyps_f:
                         for hyp in hyps:
                             print(" ".join(hyp), file=hyps_f)
-                    if epoch == 0:
+                    if epoch == 1:
                         with open(os.path.join(hyps_dir, "refs"), "w", 
                                   encoding=ENCODING) as refs_f:
                             for ref in refs:

diff --git a/tox.ini b/tox.ini
@@ -1,5 +1,5 @@
 [tox]
-envlist=py35,py36
+envlist=py35,py36,py37
 [testenv]
 deps=
 	-r{toxinidir}/test_requirements.txt