From f1cb0e4a3a0787b7a5ec2fc4fd05b018e3b5858c Mon Sep 17 00:00:00 2001 From: felix Date: Fri, 16 Oct 2026 13:05:40 +0200 Subject: [PATCH] [tests] Relax compiled detection models seg_map comparision --- tests/pytorch/test_models_detection_pt.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/pytorch/test_models_detection_pt.py b/tests/pytorch/test_models_detection_pt.py index e1ad77fd3b..6fea2e32df 100644 --- a/tests/pytorch/test_models_detection_pt.py +++ b/tests/pytorch/test_models_detection_pt.py @@ -222,7 +222,8 @@ def test_torch_compiled_models(arch_name, mock_payslip): for out_boxes, compiled_out_boxes in zip(out, compiled_out) ) assert all( - np.allclose(seg_map, compiled_seg_map, atol=1e-4) + # small increased tolerance because `fast_small` is slightly over the 1e-4 threshold + np.allclose(seg_map, compiled_seg_map, atol=2e-4) for seg_map, compiled_seg_map in zip(seg_maps, compiled_seg_maps) )