Skip to content

Commit b9e28cf

Browse files
NicolasHugfacebook-github-bot
authored andcommitted
[fbsync] add minor tolerance for flaky bounding box test (#6995)
Reviewed By: YosuaMichael Differential Revision: D41648536 fbshipit-source-id: f781555a55dce01a7c23dcbbd0502fd08a07eca3
1 parent 451399b commit b9e28cf

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

test/prototype_transforms_kernel_infos.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -372,6 +372,9 @@ def reference_inputs_resize_bounding_box():
372372
sample_inputs_fn=sample_inputs_resize_bounding_box,
373373
reference_fn=reference_resize_bounding_box,
374374
reference_inputs_fn=reference_inputs_resize_bounding_box,
375+
closeness_kwargs={
376+
(("TestKernels", "test_against_reference"), torch.int64, "cpu"): dict(atol=1, rtol=0),
377+
},
375378
test_marks=[
376379
xfail_jit_python_scalar_arg("size"),
377380
],

0 commit comments

Comments
 (0)