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, max_steps=100,
exclude=[], exclude=[],
annotating_components=[], annotating_components=[],
rehearse_components=[],
before_update=before_update, before_update=before_update,
) )

View File

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