ci(): add generated members, more lenient tests

sign-language-processing · Jan 6, 2024 · 5753240 · 5753240
1 parent b36017f
commit 5753240
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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__':