diff --git a/spacy/training/loop.py b/spacy/training/loop.py index 9368384ae..4f50d6a94 100644 --- a/spacy/training/loop.py +++ b/spacy/training/loop.py @@ -155,7 +155,7 @@ def train_while_improving( max_steps: int, exclude: List[str], annotating_components: List[str], - rehearse_components: List[str], + rehearse_components: Optional[List[str]] = [], before_update: Optional[Callable[["Language", Dict[str, Any]], None]], ): """Train until an evaluation stops improving. Works as a generator,