use torch.testing.assertclose instead to get more details about error in cis (#35659)
* use torch.testing.assertclose instead to get more details about error in cis * fix * style * test_all * revert for I bert * fixes and updates * more image processing fixes * more image processors * fix mamba and co * style * less strick * ok I won't be strict * skip and be done * up
This commit is contained in:
@@ -506,5 +506,5 @@ class Wav2Vec2ProcessorWithLMTest(unittest.TestCase):
|
||||
expected_end_tensor = torch.tensor([0.7800, 1.1000, 1.6600, 1.9200, 2.0400, 2.8000, 3.3000, 3.8800, 4.2800])
|
||||
# fmt: on
|
||||
|
||||
self.assertTrue(torch.allclose(start_times, expected_start_tensor, atol=0.01))
|
||||
self.assertTrue(torch.allclose(end_times, expected_end_tensor, atol=0.01))
|
||||
torch.testing.assert_close(start_times, expected_start_tensor, rtol=0.01, atol=0.01)
|
||||
torch.testing.assert_close(end_times, expected_end_tensor, rtol=0.01, atol=0.01)
|
||||
|
||||
Reference in New Issue
Block a user