fix: Pytests for yolo (#699)

makseq · web-flow · commit fcb19b6a2027 · 2025-01-28T14:18:17.000Z
diff --git a/label_studio_ml/examples/yolo/tests/test_keypoint_labels.py b/label_studio_ml/examples/yolo/tests/test_keypoint_labels.py
@@ -131,4 +131,4 @@ def test_keypoints_predict(client, label_config, task, expect):
     )
     assert response.status_code == 200, "Error while predict"
     data = response.json
-    compare_nested_structures(data["results"], expect)
+    compare_nested_structures(data["results"], expect, rel=1e-1)

Original file line number	Diff line number	Diff line change
`@@ -131,4 +131,4 @@ def test_keypoints_predict(client, label_config, task, expect):`
`131`	`131`	`)`
`132`	`132`	`assert response.status_code == 200, "Error while predict"`
`133`	`133`	`data = response.json`
`134`		`- compare_nested_structures(data["results"], expect)`
	`134`	`+ compare_nested_structures(data["results"], expect, rel=1e-1)`