Fix types

This commit is contained in:
Adriane Boyd 2022-10-25 16:07:54 +02:00
parent a36891ac4e
commit 037142ca22

View File

@ -155,7 +155,7 @@ class TextCategorizer(TrainablePipe):
self.model = model
self.name = name
self._rehearsal_model = None
cfg = {"labels": [], "threshold": threshold, "positive_label": None}
cfg: Dict[str, Any] = {"labels": [], "threshold": threshold, "positive_label": None}
self.cfg = dict(cfg)
self.scorer = scorer