diff --git a/tests/models/deformable_detr/test_modeling_deformable_detr.py b/tests/models/deformable_detr/test_modeling_deformable_detr.py index 589543506ee..127bf1c6e09 100644 --- a/tests/models/deformable_detr/test_modeling_deformable_detr.py +++ b/tests/models/deformable_detr/test_modeling_deformable_detr.py @@ -771,7 +771,7 @@ class DeformableDetrModelIntegrationTests(unittest.TestCase): # 3. assert equivalence # (on A10, the differences get larger than on T4) for key in cpu_outputs.keys(): - assert torch.testing.assert_close(cpu_outputs[key], gpu_outputs[key].cpu(), atol=1e-2, rtol=1e-2) + torch.testing.assert_close(cpu_outputs[key], gpu_outputs[key].cpu(), atol=1e-2, rtol=1e-2) expected_logits = torch.tensor( [