Skip to content

GitLab

Unverified Commit 647b32ab authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

add minor tolerance for flaky bounding box test (#6995)

parent 4941c6b6
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
test/prototype_transforms_kernel_infos.py
View file @ 647b32ab
...@@ -372,6 +372,9 @@ KERNEL_INFOS.extend( ...@@ -372,6 +372,9 @@ KERNEL_INFOS.extend(
sample_inputs_fn=sample_inputs_resize_bounding_box, sample_inputs_fn=sample_inputs_resize_bounding_box,
reference_fn=reference_resize_bounding_box, reference_fn=reference_resize_bounding_box,
reference_inputs_fn=reference_inputs_resize_bounding_box, reference_inputs_fn=reference_inputs_resize_bounding_box,
closeness_kwargs={
(("TestKernels", "test_against_reference"), torch.int64, "cpu"): dict(atol=1, rtol=0),
},
test_marks=[ test_marks=[
xfail_jit_python_scalar_arg("size"), xfail_jit_python_scalar_arg("size"),
], ],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment