From 25d7bb3ea65bb6f79056252584e24eb545c5eb18 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Wed, 19 Jun 2024 21:48:08 +0100 Subject: [PATCH] [Flax tests] reduce tolerance for a flax test (#8640) reduce tolerance for a flax test --- tests/schedulers/test_scheduler_flax.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/schedulers/test_scheduler_flax.py b/tests/schedulers/test_scheduler_flax.py index 2855f09f7e..d2ee7e1314 100644 --- a/tests/schedulers/test_scheduler_flax.py +++ b/tests/schedulers/test_scheduler_flax.py @@ -338,7 +338,7 @@ class FlaxDDPMSchedulerTest(FlaxSchedulerCommonTest): assert abs(result_sum - 255.0714) < 1e-2 assert abs(result_mean - 0.332124) < 1e-3 else: - assert abs(result_sum - 255.1113) < 1e-2 + assert abs(result_sum - 255.1113) < 1e-1 assert abs(result_mean - 0.332176) < 1e-3