riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

AutoformerForPrediction test add atol (#36017)

Browse Source
This commit is contained in:
ivarflakstad 2025-02-10 19:22:24 +01:00 committed by GitHub
parent be2ac0916a
commit 1feebb5b41
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/autoformer/test_modeling_autoformer.py
View File

@ -481,4 +481,4 @@ class AutoformerModelIntegrationTests(unittest.TestCase):
expected_slice = torch.tensor([3130.6763, 4056.5293, 7053.0786], device=torch_device) expected_slice = torch.tensor([3130.6763, 4056.5293, 7053.0786], device=torch_device)
mean_prediction = outputs.sequences.mean(dim=1) mean_prediction = outputs.sequences.mean(dim=1)
torch.testing.assert_close(mean_prediction[0, -3:], expected_slice, rtol=1e-1) torch.testing.assert_close(mean_prediction[0, -3:], expected_slice, rtol=1e-1, atol=1e-1)