Fixing mypy errors

Author: Dmitryv-2024

autointent/modules/scoring/_dnnc/dnnc.py

@@ -252,7 +252,7 @@ def _get_cross_encoder_scores(self, utterances: list[str], candidates: list[list
 
         flattened_cross_encoder_scores: npt.NDArray[np.float64] = self.model.predict(flattened_text_pairs)
         return [
-            flattened_cross_encoder_scores[i : i + self.k].tolist()
+            flattened_cross_encoder_scores[i : i + self.k].tolist()  # type: ignore[misc]
             for i in range(0, len(flattened_cross_encoder_scores), self.k)
         ]
 

autointent/modules/scoring/_knn/weighting.py

@@ -95,5 +95,5 @@ def to_onehot(labels: NDArray[Any], n_classes: int) -> NDArray[Any]:
     new_shape = (*labels.shape, n_classes)
     onehot_labels = np.zeros(shape=new_shape)
     indices = (*tuple(np.indices(labels.shape)), labels)
-    onehot_labels[indices] = 1
+    onehot_labels[indices] = 1  # type: ignore[index]
     return onehot_labels