From 39163f4e7a9648adc05f36e06dfe48f324861454 Mon Sep 17 00:00:00 2001 From: Larisa Markeeva Date: Mon, 2 Feb 2026 09:54:16 -0800 Subject: [PATCH] Fix flaky `losses_test` Increase tolerance in `test_hint_loss` from `rtol=1e-4` to `rtol=1e-3` to fix flaky test failures. PiperOrigin-RevId: 864401872 --- clrs/_src/losses_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clrs/_src/losses_test.py b/clrs/_src/losses_test.py index 44a181fa..7a1e50e6 100644 --- a/clrs/_src/losses_test.py +++ b/clrs/_src/losses_test.py @@ -159,7 +159,7 @@ def test_hint_loss(self, algo): lengths=full_sample.features.lengths, nb_nodes=nb_nodes, ) - np.testing.assert_allclose(chunk_hint_loss, full_hint_loss, rtol=1e-4) + np.testing.assert_allclose(chunk_hint_loss, full_hint_loss, rtol=1e-3) if __name__ == '__main__':