From 04632be5c6f65c95cf77caab8ceef954fc11af38 Mon Sep 17 00:00:00 2001 From: BrunoLiegiBastonLiegi Date: Fri, 22 Nov 2024 16:05:33 +0100 Subject: [PATCH] fix: it's starting to get frustrating --- tests/test_models_interfaces.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models_interfaces.py b/tests/test_models_interfaces.py index 4673511..13dd779 100644 --- a/tests/test_models_interfaces.py +++ b/tests/test_models_interfaces.py @@ -200,7 +200,7 @@ def backprop_test(frontend, model, data, target): grad = train_model(frontend, model, data, target) _, loss_trained = eval_model(frontend, model, data, target) assert grad < 1e-2 - assert loss_untrained >= loss_trained + assert round(float(loss_untrained), 6) >= round(float(loss_trained), 6) # in some (unpredictable) cases the gradient and loss # start so small that the model doesn't do anything # fixing the seed doesn't fix this on all the platforms