Skip to content

Commit ada60c9

Browse files
committed
Fix mypy
1 parent b2b9d4c commit ada60c9

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

dreadnode/scorers/classification.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import typing as t
22

3-
from transformers import pipeline # type: ignore[import-not-found]
3+
from transformers import pipeline
44

55
from dreadnode.meta import Config
66
from dreadnode.metric import Metric

dreadnode/scorers/similarity.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from nltk.tokenize import word_tokenize # type: ignore[import-untyped]
77
from nltk.translate.bleu_score import sentence_bleu # type: ignore[import-untyped]
88
from rapidfuzz import distance, fuzz, utils
9-
from sentence_transformers import SentenceTransformer, util # type: ignore # noqa: PGH003
9+
from sentence_transformers import SentenceTransformer, util
1010
from sklearn.feature_extraction.text import TfidfVectorizer # type: ignore[import-untyped]
1111
from sklearn.metrics.pairwise import ( # type: ignore # noqa: PGH003
1212
cosine_similarity as sklearn_cosine_similarity,

0 commit comments

Comments
 (0)