make rehearse_components optional in functions

This commit is contained in:
thomashacker 2023-01-19 21:19:22 +01:00
parent e9057d2cc6
commit 209cbe196b

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: List[str], rehearse_components: Optional[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,