SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 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
1 changed files with 1 additions and 1 deletions
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)
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)