We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 292ac59 commit e047f23Copy full SHA for e047f23
src/autointent/modules/scoring/_bert.py
@@ -151,7 +151,7 @@ def fit(
151
152
self._validate_task(labels)
153
154
- self._tokenizer = transformers.AutoTokenizer.from_pretrained(self.classification_model_config.model_name) # type: ignore[no-untyped-call]
+ self._tokenizer = transformers.AutoTokenizer.from_pretrained(self.classification_model_config.model_name)
155
self._model = self._initialize_model()
156
tokenized_dataset = self._get_tokenized_dataset(utterances, labels)
157
self._train(tokenized_dataset)
0 commit comments