torch.testing.assert_allclose vs torch.testing.assert_close migration change (!50) · Merge requests · Automatic Text Recognition / PyLaia · GitLab

Yoann Schneider requested to merge fix-test-failure into master Dec 19, 2023

The migration was done in https://gitlab.teklia.com/atr/pylaia/-/commit/3c07ad42e16dc9197278abfdefee3e012c4bddfc#32c445e1c8e4b33a24298741d5cb50e9bf76b57d_17_16. However, sometimes this test failed because it was trying to compare nan values. While assert_allclose supported it by default, assert_close doesn't.