fix test

Author: riapush

autointent/_callbacks/tensorboard.py

@@ -16,7 +16,7 @@ def __init__(self) -> None:
         Raises an ImportError if neither are installed.
         """
         try:
-            from torch.utils.tensorboard import SummaryWriter  # type: ignore[attr-defined]
+            from torch.utils.tensorboard import SummaryWriter
 
             self.writer = SummaryWriter
         except ImportError:

autointent/modules/scoring/_cnn/cnn.py

@@ -88,7 +88,7 @@ def fit(self, utterances: list[str], labels: ListOfLabels) -> None:
         self._multilabel = isinstance(labels[0], (list, np.ndarray)) # noqa: UP038
 
         if self._multilabel:
-            self._n_classes = len(labels[0])
+            self._n_classes = len(labels[0]) #type: ignore[arg-type]
         else:
             self._n_classes = len(set(labels)) if labels else 0
 

autointent/modules/scoring/_cnn/textcnn.py

@@ -24,7 +24,7 @@ def __init__(
 
         if pretrained_embs is not None:
             _, embed_dim = pretrained_embs.shape
-            self.embedding = nn.Embedding.from_pretrained(pretrained_embs, freeze=True)
+            self.embedding = nn.Embedding.from_pretrained(pretrained_embs, freeze=True) # type: ignore[no-untyped-call]
         else:
             self.embedding = nn.Embedding(
                 num_embeddings=vocab_size,

tests/modules/scoring/test_cnn.py

@@ -73,7 +73,7 @@ def test_cnn_cache_clearing(dataset):
     assert not hasattr(scorer, "_model") or scorer._model is None
 
     # После очистки кэша предсказания должны вызывать ошибку
-    with pytest.raises(ValueError, match="Model not trained. Call fit() first."):
+    with pytest.raises(ValueError, match=r"Model not trained\. Call fit\(\) first\."):
         scorer.predict(test_data)