• moto's avatar
    Use torch.testing.assert_allclose (#513) · 5f5df1d6
    moto authored
    * grep -l 'torch.allclose' -r test | xargs sed -i 's/assert torch.allclose/torch.testing.assert_allclose/g'
    
    * grep -l 'torch.allclose' -r test | xargs sed -i 's/self.assertTrue(torch.allclose(\(.*\)))/torch.testing.assert_allclose(\1)/g'
    
    * Fix missing atol/rtol, wrong shape, argument order. Remove redundant shape assertions
    5f5df1d6
test_functional.py 13.5 KB