Fix mypy and tests

This commit is contained in:
thomashacker 2023-01-20 10:18:26 +01:00
parent 209cbe196b
commit 05a2f560ad
2 changed files with 2 additions and 1 deletions

View File

@ -1144,6 +1144,7 @@ def test_training_before_update(doc):
max_steps=100,
exclude=[],
annotating_components=[],
rehearse_components=[],
before_update=before_update,
)

View File

@ -155,7 +155,7 @@ def train_while_improving(
max_steps: int,
exclude: List[str],
annotating_components: List[str],
rehearse_components: Optional[List[str]] = [],
rehearse_components: List[str],
before_update: Optional[Callable[["Language", Dict[str, Any]], None]],
):
"""Train until an evaluation stops improving. Works as a generator,