diff --git a/pyproject.toml b/pyproject.toml
index 1a014dd..83f49c5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,7 +36,10 @@ disable = [
 ]
 
 [tool.pylint.typecheck]
-generated-members = ["torch.*"]
+generated-members = [
+    "torch.*",
+    "torch.nn.functional.*",
+]
 
 [tool.setuptools]
 packages = [
diff --git a/signwriting_evaluation/metrics/test_clip.py b/signwriting_evaluation/metrics/test_clip.py
index e74d81d..2fd6ddf 100644
--- a/signwriting_evaluation/metrics/test_clip.py
+++ b/signwriting_evaluation/metrics/test_clip.py
@@ -11,7 +11,7 @@ def test_score(self):
         reference = "M519x534S37900497x466S3770b497x485S15a51491x501S22f03481x513"
         score = self.metric.score(hypothesis, reference)
         self.assertIsInstance(score, float)  # Check if the score is a float
-        self.assertAlmostEqual(score, 0.96179789)
+        self.assertAlmostEqual(score, 0.96179, places=4)
 
 
 if __name__ == '__main__':