diff --git a/spacy/ty.py b/spacy/ty.py index 8f2903d78..09be540e8 100644 --- a/spacy/ty.py +++ b/spacy/ty.py @@ -1,5 +1,7 @@ from typing import TYPE_CHECKING from typing import Optional, Any, Iterable, Dict, Callable, Sequence, List + +from spacy.language import Language from .compat import Protocol, runtime_checkable from thinc.api import Optimizer, Model @@ -32,7 +34,7 @@ class InitializableComponent(Protocol): def initialize( self, get_examples: Callable[[], Iterable["Example"]], - nlp: Iterable["Example"], + nlp: Language, **kwargs: Any ): ...