renaming

Author: SeBorgey

autointent/modules/scoring/__init__.py

@@ -4,7 +4,7 @@
 from ._linear import LinearScorer
 from ._mlknn import MLKnnScorer
 from ._sklearn import SklearnScorer
-from ._transformer import BertScorer
+from ._bert import BertScorer
 
 __all__ = [
     "BertScorer",

…tointent/modules/scoring/_transformer.py autointent/modules/scoring/_bert.pyautointent/modules/scoring/_transformer.py renamed to autointent/modules/scoring/_bert.py autointent/modules/scoring/_transformer.py renamed to autointent/modules/scoring/_bert.py

@@ -112,13 +112,13 @@ def fit(
 
         use_cpu = hasattr(self.model_config, "device") and self.model_config.device == "cpu"
 
-        def tokenize_function(examples: dict[str, Any]) -> dict[str, Any]:  # type: ignore[no-any-return]
+        def tokenize_function(examples: dict[str, Any]) -> dict[str, Any]:
             return self._tokenizer(
                 examples["text"],
                 padding=self.tokenizer_config.padding,
                 truncation=self.tokenizer_config.truncation,
                 max_length=self.tokenizer_config.max_length,
-            )
+            )  # type: ignore[no-any-return]
 
         dataset = Dataset.from_dict({"text": utterances, "labels": labels})
         tokenized_dataset = dataset.map(tokenize_function, batched=True)